关于国内conda安装cuda11.6+pytorch的那些事。

因为一些众所周知的原因，你安装cuda11.6+pytorch的时候是会很蛋疼的。于是我们需要修改一下cuda的软件源。

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
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/bioconda/
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/conda-forge/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge
conda config --set show_channel_urls true

以上命令复制粘贴即可。

需要注意的一点是，官方安装pytorch给的命令是

conda install pytorch torchvision torchaudio pytorch-cuda=11.6 -c pytorch -c nvidia

我们需要修改一下，去掉 -c pytorch -c nvidia

conda install pytorch torchvision torchaudio pytorch-cuda=11.6

这就搞定了。