在anaconda中安装pytorch的GPU版本

2024-01-08 17:21:07

本文前提：

1.你已经下载好了anaconda,最好是新建一个虚拟环境来安装pytorch的GPU版本，并且设置了国内镜像源；
2.了解自己电脑对应的cuda版本，可通过nvidia-smi，并下载好了cuda。

在这里插入图片描述

安装pytorch的GPU版本

到官网中https://pytorch.org/首页向下翻，找到以下内容，然后根据自己的cuda版本复制相应的命令。安装包相对较大，请耐心等待。
在这里插入图片描述
安装完成后，通过conda list可查看已下载的依赖：
在这里插入图片描述
g.cn/direct/da72a6920a9f432eaebf845351e7dc96.png)

gpu版本能够看到cuda相应的依赖包，而以下是非gpu版本：
在这里插入图片描述
你也可以通过查看以下的输出查看是否安装成功：

torch.__version__
torch.cuda.is_available()
torch.cuda.get_device_name(0)

在这里插入图片描述

文章来源:https://blog.csdn.net/qq_45939398/article/details/135458275
本文来自互联网用户投稿，该文观点仅代表作者本人，不代表本站立场。本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。