博客

安装CUDA、CUDnn那些就不赘述，以及下载的pytorch要和自己CUDA版本匹配，比如我的是Ubuntu，CUDA版本为11.6，直接选择对应的。

在安装过，别的提到的方法都试过，比如：

添加清华镜像链接

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

但是下载快完成的时候还是会报错。

反复试了下了六次，最后终于找到一个正确的方法。即直接在pytorch官网复制的下载连接后面加入清华镜像，即下面这一句：

-i https://pypi.tuna.tsinghua.edu.cn/simple

比如我原来的下载链接是：

pip3 install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu116

添加后就成了：

pip3 install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu116 -i https://pypi.tuna.tsinghua.edu.cn/simple

刚下会报黄，但是可以下载成功。用了别的什么添加镜像链接方法，都不管用。还有一个小tip，如果用conda下不成，就用pip下。