使用conda安装GPU版本的Pytorch

问题

在python3.7的环境下，使用如下命令安装pytorch1.11，结果安装成CPU版本 - -||

conda install pytorch==1.11.0 torchvision==0.12.0 torchaudio==0.11.0 cudatoolkit=11.3

CPU版本的体现一
在命令行python解释器下输入以下指令

CPU版本的体现二
在conda安装包中查看pytorch版本

conda list | grep torch

解决方法

1. 查看镜像源中是否有对应GPU版本的Pytorch

我需要安装python3.7-pytorch1.11-cuda11.3版本，以清华源为例，下图红框部分即为我需要安装的Pytorch版本。

补充：若cuda11.4版本的话安装cuda11.3版本也可
若该镜像源中没有需要安装版本的pytorch，可以查看其他镜像源例如阿里云、豆瓣，或者安装其他版本的Pytorch

2.安装GPU版本Pytorch

首先卸载原来安装的Pytorch，我这里使用conda命令安装Pytorch，于是使用conda命令卸载

conda uninstall pytorch

执行如下命令重新安装Pytorch

 conda install pytorch==1.11.0 torchvision==0.12.0 torchaudio==0.11.0 cudatoolkit=11.3 -c https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/linux-64/ 

注意这里相对于最开始安装Pytorch的conda命令，添加了 -c https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/linux-64/ 其中-c表示channel，后面跟上的url表示conda将从该url下安装pytorch

成功安装GPU版本的Pytorch! ^ _ ^

其他想法

这里提供一个可能可以一劳永逸的方法，即不需要每次用conda install 都加上 channel的方法。
修改一下~/.condarc文件，将包含所需要安装的软件包的镜像源提前试试。

如上图所说，我把提供GPU版本的channel红框部分提前放在导致我装CPU版本的conda-forge之前了。

但conda貌似也不会按照condarc配置文件中的channel顺序装包 - -||

参考文献

1. 查看是否安装GPU版本的Pytorch
2. 卸载Pytorch
3. conda安装GPU版pytorch，结果却是cpu版本