torch各个版本镜像_如何解决在cuda上安装torch后torch.cuda.is_available()返回False

torch各个版本镜像_如何解决在cuda上安装torch后torch.cuda.is_available()返回False_第1张图片

课题组的服务器最近又炸了,不得已换一台稳定点的服务器,让师弟开了账号开始部署自己的代码环境。装了anaconda之后,开始安装用的最多的torch。

初安装:使用清华镜像下载安装torch、torchvision,代码如下

conda install pytorch torchvision cudatoolkit=10.1 -c pytorch

中间遇到了HTTP ERROR的问题,设置了几条清华镜像,之后重复上述命令仍然无法解决该网络问题。

torch各个版本镜像_如何解决在cuda上安装torch后torch.cuda.is_available()返回False_第2张图片

最后发现设置了镜像之后需要修改代码如下,才能用镜像地址下载,否则仍然是default地址下载:

conda install pytorch torchvision cudatoolkit=10.1

安装完验证是否安装成功:

ebcb187227acaa4365b7c1eb6404b289.png

发现import是没有问题的,但是在运行torch.cuda.is_available()之后返回值却是False!!!

不要慌,开始百度解决方案,经过对多个方案的分析,最后选择了以下方法,一次成功。

先在这里下载对应cuda版本的torch和torchvision“轮子文件”(这名字我自己取的,无参考文献无理论支撑无实际意义,三无产品)

我下载的是这两个文件:

torch各个版本镜像_如何解决在cuda上安装torch后torch.cuda.is_available()返回False_第3张图片

完了之后通过WinSCP或者你常用的其他文件传输协议把这两个文件传输到服务器上,anaconda3/pkgs文件夹下:

torch各个版本镜像_如何解决在cuda上安装torch后torch.cuda.is_available()返回False_第4张图片

接下来,在该文件夹下打开终端,依次运行以下代码:

pip install 文件名1
pip install 文件名2

因为下载的文件版本不一样,所以这里用文件名1/文件名2来代替,我运行的是以下2个命令,可以看到两个命令分别成功地安装上了torch 和torchvision:

torch各个版本镜像_如何解决在cuda上安装torch后torch.cuda.is_available()返回False_第5张图片

torch各个版本镜像_如何解决在cuda上安装torch后torch.cuda.is_available()返回False_第6张图片

最后验证一下,在重装torch 和torchvision之后,torch.cuda.is_available()的返回值。

37cb2b6ace6fc42558f922e94c561e50.png

成功撒花!*★,°*:.☆( ̄▽ ̄)/$:*.°★* 。

有不明白的可以评论交流~

你可能感兴趣的:(torch各个版本镜像)