因为一些众所周知的原因,你安装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
这就搞定了。