之前已经装过CUDA和cudNN了

1.查看CUDA版本：nvcc -V

2.查看cuDNN版本：

找到自己安装的CUDA的地方（我在D盘），找到cudnn_version.h文件，记事本打开，如图，我的是8.1.0版本

D:\ProgramFiles\NVIDIA-CUDA\NVIDIA GPU Computing Toolkit\include\cudnn_version.h

3.查询应该安装的pytorch版本

进入pytorch官网（Start Locally | PyTorch）

根据CUDA选择版本，先看红色框有没有对应版本

如果没有，就进入绿色框的连接寻找，我的是11.1版本，可以选torch版本如下所示：

 我最终选择1.9.0版本。

安装：复制pip那行

pip install torch==1.9.0+cu111 torchvision==0.10.0+cu111 torchaudio==0.9.0 -f https://download.pytorch.org/whl/torch_stable.html

 测试：

python

import torch
print(torch.__version__)  #注意是双下划线
print(torch.version.cuda)
print(torch.cuda.is_available())
print(torch.cuda.get_device_name())

下图是安装界面和测试：（我新建了一个环境，conda create -n XXXXX python=3.6）