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