1.使用官网推荐的方法(不推荐):官网链接
在命令行输入
conda install pytorch torchvision cudatoolkit=9.0 -c pytorch
会发现pytorch下载得非常慢,因为它是从国外的服务器下载,而且到最后可能还会超时,前功尽弃。
而且即使添加了国内的源(如清华源),conda也不一定会解析到那里去。
这个时候我们可以手动从清华源下载,再安装。
2.使用清华镜像下载pytorch,快速离线安装
在https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/文件夹下,选择linux-64。
然后找到pytorch,图中是一部分pytorch链接,
然后你就可以根据python版本,cuda版本选择你需要装的pytorch了。鼠标右键选择复制链接地址(例如https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/linux-64/pytorch-1.0.1-cuda90py36h8b0c50b_0.tar.bz2)
然后命令行输入: wget 链接地址
速度飞起哦
下载完成之后
在同一文件夹下运行:conda install pytorch-1.0.1-cuda90py36h8b0c50b_0.tar.bz2
再安装torchvision : conda install torchvision
至此pytorch安装完成,再装上cuda等其他模块就可以开始你的入坑之旅了。