记我第二次安装pytorch-gpu
最近因为手贱不小心把anaconda的pytorch-gpu的虚拟环境给删了,超操气。但是没办法还是得重新装回去。还记得之前第一次装就装了一天才不明不白的装上,还好这次运气不错,找对了方法,很快就装上了。下面就来介绍一下我的安装方法。
首先,因为我已经装过一次所以已经有cuda 和 cudnn 所以这次就不用再装,还没装的朋友们别急参考这篇文章(23条消息) Cuda和cuDNN安装教程(超级详细)_jhsignal的博客-CSDN博客_cudnn安装
一步一步来。
其次,anaconda的下载通道我用的是清华镜像,你可以用一下命令添加:
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
进入要配置的虚拟环境:activate XXXXX
然后就是我这次的经验积累了。上官网直接copy conda方式的安装命令(我没管合不合版本,直接就是执行命令),执行命令前切换到自己的流量网络到电脑!!!(用校园网不可能完成安装!!!是这次最大的发现)
使用conda指令去安装会方便很多,因为它会自动帮你下载相关的依赖,不用你一个个pip,对于一些清华镜像里面有的资源它会自动从清华镜像里头下载,对于一些清华镜像里没有的比如pytorch还得从官网的源下载,所以从官网源下载的话会比较慢(校园网的话直接下载不了),这里有vpn的同学可以打开vpn会稍微快一些。(当然只在下载的时候,而且是在下载外国源的时候才打开vpn,否则在执行命令之初用vpn的话会直接因为连不上清华镜像而执行失败)
执行后命令台会问你是否安装以上相关的包,这个时候看准了,pytorch应该是cuXXX的而不是cpu
如果是cpu版本的话n+enter。再次检查一下网络是否是手机热点网络,以及非vpn(vpn是在开始下载官方源后再开),再次执行官网配置命令。
下载过程中如果意外中断,没关系,再来几次,之前下载好的不会conda会跳过避免重复下载。所有内容下载完毕后就会自动安装。到这里你的pytorch-gpu就安装完成了。下面验证一下安装吧。
输入python,然后import torch,然后print(torch.cuda.is_available())
如果打印结果为Ture,那就恭喜你,你的pytorch-gpu版已经安装好了,否则就没安装成功,需要重装(如果方便的话删除虚拟环境,重新再来一次)