使用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顺序装包 - -||
参考文献
- 查看是否安装GPU版本的Pytorch
- 卸载Pytorch
- conda安装GPU版pytorch，结果却是cpu版本