anaconda安装pytorch-GPU版本(python3.7)

anaocnda安装pytoch-GPU版本(解决:默认CPU版本安装)



一、总体步骤规划

1、确定电脑的cuda版本。
2、确定python的版本。
3、确定要下载的pytorch版本,torchaudio版本,以及torchvision版本
4、下载对应版本的whl镜像安装文件。
5、在anaconda prompt中激活环境,按顺序安装对应的whl文件。
6、测试是否安装成功。

注:本次教程下载torch1.10.1版本,cuda10.2版本,python3.7.4版本。

二、查看cuda版本

快捷键win+R,输入cmd,回车。
在命令行中输入nvidia-smi
在这里插入图片描述
可以看到,本机的CUDA Version是11.6版本,所以在选择torch+cuda版本时,要注意,必须小于11.6.

三、确定并下载pytorch、torchaudio以及torchversion对应版本安装文件

注意:以python3.7版本,window系统为例。

1、确定torch版本及下载安装文件

打开网址https://download.pytorch.org/whl/cu102
在这里插入图片描述
点击torch,可以看到很多版本torch的whl安装文件。
在这里插入图片描述
按快捷键ctrl+f,搜索cu102-cp37(表示cuda10.2版本,python3.7版本),寻找合适的安装文件。
这里我们选择torch1.10.1版本,cuda10.2版本,python3.7版本
在这里插入图片描述
点击下载即可。
注:如果下载速度过慢,建议改换手机热点进行下载,速度很快。

2、确定torchvision版本及下载安装文件

打开网址https://pytorch.org/get-started/previous-versions/

找到下载的torch版本对应的torchvision版本。
如torch1.10.1对应的torchvision版本为0.11.2。
在这里插入图片描述
打开网址https://download.pytorch.org/whl/cu102选择torchversion
在这里插入图片描述
点击进入,同理ctrl+f搜索“cu102-cp37”,点击下载该版本的torchversion
在这里插入图片描述

3、确定torchaudio的版本及下载安装文件

与第二步同理,确定torchaudio的版本为0.10.2
在这里插入图片描述
进入torchaudio
在这里插入图片描述
找到下载的文件。
在这里插入图片描述

点击下载即可。

四、在anaconda中安装对应whl文件

将三个文件放到一个文件夹下。如D:/pytorch_whl
在这里插入图片描述
在anaconda prompt中激活或创建一个虚拟环境
在这里插入图片描述
在这里插入图片描述
依次安装三个whl文件。注意文件的绝对路径。以刚才的安装路径D:\pytorch_whl 为例,依次按下面顺序输入

pip install D:\pytorch_whl\torch-1.10.1+cu102-cp37-cp37m-win_amd64.whl
pip install D:\pytorch_whl\torchaudio-0.10.1+cu102-cp37-cp37m-win_amd64.whl
pip install D:\pytorch_whl\torchvision-0.11.2+cu102-cp37-cp37m-win_amd64.whl

解析:pip install 绝对路径\文件名

等待安装完成即可。


五、测试安装成功

在虚拟环境下打开python,进入python编程环境后输入import torch 回车后输入torch.cuda.is_available()。如果返回True则安装成功。
在这里插入图片描述
在这里插入图片描述

有问题可以查看本人的其他文章,如虚拟环境的创建等。
或者评论询问。

谢谢!!!