问题

在使用pytorch官网的指令安装pytorch时,发现安装的速度很慢,官网的安装指令如下(以1.1.0为例):

1
conda install pytorch==1.1.0 torchvision==0.3.0 cudatoolkit=10.0 -c pytorch

解决方案

这是因为下载的时候使用的是pytorch官网的源,我们修改源就行了(这里使用的清华源)。执行下面代码修改源:

1
2
3
4
5
6
7
8
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
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/conda-forge/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/msys2/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/bioconda/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/menpo/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/fastai/

然后执行下面命令安装:

1
conda install pytorch==1.1.0 torchvision==0.3.0 cudatoolkit=10.0 

这里去掉了 -c pytorch,因为-c pytorch表明使用的channelspytorch,去掉这个才能使用修改的清华源。

注意

如果安装的时候报错,将.condarc中有一行default去掉