anaconda环境下载安装GPU版本pytorch时遇到的问题及解决方法(cuda11.3)

毕设要使用pytorch，因为之前下载过tensoflow所以知道要找对应版本，就先看了一下自己需要下载的版本：

查询链接：https://pytorch.org/get-started/locally/#no-cuda-1

从这个图可以看到我要下载的是cuda11.3。
可以直接在anaconda prompt激活要安装pytorch的虚拟环境，通过以下语句安装cuda及cudnn。

# 安装CUDA
conda install cudatoolkit=11.3 # 指定版本
# 安装cudnn，如果不指定版本，在安装CUDA之后，会自动匹配对应版本的cudnn安装
conda install cudnn

出现问题：

安装完cuda和cudnn开始安装pytorch，刚开始我直接复制了上面图片中的语句：conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch
但是安装完成使用

torch.cuda.is_available()

验证时会出现False。
差了百度之后发现使用这个语句默认安装cpu版本的pytorch，要下载gpu版本pytorch需要使用pip下载。

也就是：

但是直接使用：

pip3 install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu113

会因为torch文件下载太慢导致下载失败。

---

解决方案：

可以直接进入上述语句后缀的网站下载相应torch、torchvision、torchaudio的whl文件：

https://download.pytorch.org/whl/cu113

查了很多帖子发现之前的官网语句直接标明了需要下载的版本，但是现在官网给的语句没有版本号。
但是我发现，使用以上语句下载时后面会直接跟上版本号：
在网站中找到对应版本号的whl文件并且下载

下载完成后在cmd（不是anaconda prompt，在这里还是回超时）进入whl文件的下载目录，并运行：

pip install torch-1.11.0+cu113-cp39-cp39-win_amd64.whl

注意：这条语句会把pytorch安装到计算机默认的python环境中，也就是环境变量里在path中的文件位置。所以想要把pytorch安装到你要用的python环境（不是默认python环境）中有两个办法：
一、在默认python环境中安装好pytorch后，在默认python环境/Lib/site-packages中找到pytorch的两个文件夹，把他们剪切到你要用的python环境/Lib/site-packages中。
二、把环境变量path中原有的python路径删掉，添加你要使用的python路径。

安装完成后回到anaconda prompt，激活虚拟环境并再次使用：

pip3 install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu113
或
pip3 install torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu113

之前下载失败是由于torch文件太大了，而torchvision和torchaudio都比较小，所以可以正常下载。

完成之后使用

python
import torch
torch.cuda.is_available()

如果出现True，那么恭喜你安装成功辣！！！