解决pytorch官网安装慢的办法

以下方法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下载太慢的解决办法

你可能感兴趣的:(python,pytorch)