以下方法win平台和linux平台应该都可以。
激活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 --set show_channel_urls yes
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
比如我的电脑cuda版本是9.0,所以安装的是pytorch=1.1.0,torchvision=0.3.0,官网的命令是:
conda install pytorch==1.1.0 torchvision==0.3.0 cudatoolkit=9.0 -c pytorch
将官网命令替换为:
conda install pytorch==1.1.0 torchvision==0.3.0 cudatoolkit=9.0
还有另一种方法参见:pytorch下载太慢的解决办法