anaconda安装torch_如何解决在cuda上安装torch后torch.cuda.is_available()返回False

anaconda安装torch_如何解决在cuda上安装torch后torch.cuda.is_available()返回False_第1张图片

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

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

conda install pytorch torchvision cudatoolkit=10.1 -c pytorch

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

anaconda安装torch_如何解决在cuda上安装torch后torch.cuda.is_available()返回False_第2张图片

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

conda install pytorch torchvision cudatoolkit=10.1

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

cb1a5df97c4c76d55f7b16b12147be5c.png

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

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

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

我下载的是这两个文件:

anaconda安装torch_如何解决在cuda上安装torch后torch.cuda.is_available()返回False_第3张图片

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

anaconda安装torch_如何解决在cuda上安装torch后torch.cuda.is_available()返回False_第4张图片

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

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

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

anaconda安装torch_如何解决在cuda上安装torch后torch.cuda.is_available()返回False_第5张图片

anaconda安装torch_如何解决在cuda上安装torch后torch.cuda.is_available()返回False_第6张图片

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

34b61c83336e8590143871e096ba6d12.png

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

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

你可能感兴趣的:(anaconda安装torch,cuda,安装,cuda,镜像,cuda安装,torchvision安装)