win10下,conda安装GPU版pytorch最新版本:cuda11.7,遇到的坑

2022.11.19

这只是我尝试成功的方法,大家可以试一下,但不一定保证有效

pytorch安装最坑的是版本匹配的问题,所有包都安装了,但就是因为版本不匹配,或者是,不同的包是通过不同的源下载的,就检测不到,报各种奇怪的错误。

这里不使用镜像源,在pytorch官网下载,但需要开代理,否则没速度。

这里使用pytorch官网的命令:

conda install pytorch torchvision torchaudio pytorch-cuda=11.7 -c pytorch -c nvidia

这里需要科\学上网开代理,速度还是可以的,除了几个很大的包之外,基本都能下载。

手动下载

最后只剩这2个比较大的包,使用conda下载不下来,经常中断退出,我们选择手动下载

nsight-compute-2022.3.0.22-0.tar.bz2
pytorch-1.13.0-py3.9_cuda11.7_cudnn8_0.tar.bz2

其对应的地址分别为

https://conda.anaconda.org/nvidia/win-64/nsight-compute-2022.3.0.22-0.tar.bz2

https://conda.anaconda.org/pytorch/win-64/pytorch-1.13.0-py3.9_cuda11.7_cudnn8_0.tar.bz2

点击链接即可在浏览器下载。

下载之后,将文件复制到自己的Anaconda的pkgs目录(Anaconda\pkgs)下,我这里是在D盘。

然后在conda终端,分别输入如下安装命令即可,注意要使用文件的绝对路径。

conda install --use-local D:\Anaconda\pkgs\pytorch-1.13.0-py3.9_cuda11.7_cudnn8_0.tar.bz2
conda install --use-local D:\Anaconda\pkgs\nsight-compute-2022.3.0.22-0.tar.bz2

安装

上面2个包安装完毕之后,再次输入官网的命令。

conda install pytorch torchvision torchaudio pytorch-cuda=11.7 -c pytorch -c nvidia

即可成功安装。

报错

如果报错提示某些文件缺失,字节数不够,或报错。可使用同样的方法手动下载:

例如,我这里提示libcusolver-dev-11.4.1.48-0中某个文件字节数不够,在网站https://anaconda.org/ 中可搜索libcusolver即可找到相应的包,https://anaconda.org/nvidia/libcusolver-dev/11.4.1.48/download/win-64/libcusolver-dev-11.4.1.48-0.tar.bz2然后手动点击,在浏览器下载,再复制到Anaconda\pkgs目录下,并解压替换之前不全的文件(夹),再次执行官网命令。

conda install pytorch torchvision torchaudio pytorch-cuda=11.7 -c pytorch -c nvidia

重复以上步骤直到安装成功即可。

验证

最后验证,在conda终端输入python,进入python环境。

import torch
print(torch.cuda.is_available())	#True
posted @ 2022-11-19 10:21  FSWLY  阅读(3773)  评论(0编辑  收藏  举报