Python下载PyTorch时，cudatoolkit包下载极慢

Background

使用以下命令下载 PyTorch 时，发现其中 cudatoolkit 包的下载速度极慢：

conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch

Solution

配置清华镜像源：

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/linux-64/
conda config --add https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/linux-64/

并去掉 PyTorch 安装命令中，最后的 -c pytorch：

conda install pytorch torchvision torchaudio cudatoolkit=10.2