Pytorch作为主流的深度学习框架,已经成为了许多同学们干活炼丹的标配,当我们分配到一个服务器的个人账号,可能环境里没有预装好pytorch或版本过低,这时候需要自己重新安装。
最常用的安装方法是通过conda来安装,在pytorch的官网上找到你需要的版本,并找到下载命令。
比如我要下载linux系统下的1.0.1版本,选择cuda9.0的GPU版,对应的命令为
conda install pytorch==1.0.1 torchvision==0.2.2 cudatoolkit=9.0 -c pytorch
然后可能速度会特别的慢,因为是从外网下载,尤其是大文件,中途很可能会超时而中断。这里提供两种解决方案:
方案一:换国内镜像源(清华源)
依次输入以下命令,将清华源添加进conda的设置里
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
conda config --set show_channel_urls yes
此时重新试一下安装的命令,注意,要