【linux+cuda11.4安装Pytorch-GPU】找不到包PackagesNotFoundError | 错下载为CPU版本Pytorch

linux+cuda11.4安装Pytorch-GPU，一系列报错且安装了CPU版本

一、安装命令

conda install pytorch torchvision torchaudio cudatoolkit=11.4 -c pytorch -c nvidia

使用官方源速度慢且可能出现找不到包

PackagesNotFoundError: The following packages are not available from current channels:
  - cudatoolkit=11.4

二、换源

方法一

用add命令添加源

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 --set show_channel_urls yes
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/linux-64/

方法二

直接修改.condarc文件

channels:
  - defaults
show_channel_urls: true
channel_alias: https://mirrors.tuna.tsinghua.edu.cn/anaconda
default_channels:
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/r
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/pro
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/msys2
custom_channels:
  conda-forge: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
  msys2: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
  bioconda: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
  menpo: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
  pytorch: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
  simpleitk: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud

• 注意 https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/r 经常404，必要时可以删掉
• 由于更新过快难以同步，清华源不同步pytorch-nightly, pytorch-nightly-cpu, ignite-nightly这三个包
• 运行 conda clean -i 清除索引缓存

custom_channels中的源使用时注意要加 -c pytorch，pytorch才会替换

pytorch/linux-64::pytorch-1.13.0-py3.7_cpu_0

但channels下的url可以不指定-c pytorch，下载包时会自动搜索

总结

方法一不用加-c pytorch；方法二必须加-c pytorch。

三、成功安装后发现是CPU版本

conda list或者conda install时发现pytorch的版本为cpu

pytorch-1.13.0             |      py3.7_cpu_0        84.3 MB  pytorch
pytorch-mutex-1.0          |              cpu           3 KB  pytorch
torchaudio-0.13.0          |         py37_cpu         6.4 MB  pytorch
torchvision-0.14.0         |         py37_cpu        25.2 MB  pytorch

原因如下
未正确使用源或源中根本没有满足 python3.7 + cudatoolkit11.4 + pytorch1.13 的版本。
因此，要先在源下找到并指定好版本下载
果然在 https://mirrors.bfsu.edu.cn/anaconda/cloud/pytorch/linux-64/ （清华源有时候没有新版本会自动到北京外国语源中搜索）中没找到pytorch-1.13.0-py3.7_cuda11.7使用

conda install pytorch=1.12 torchvision torchaudio cudatoolkit=11.3 -c pytorch