一定保证版本要对应

1.首先是有N卡，然后保证安装好Nvidia驱动程序后，安装CUDA

    可参考大神博客（here_1）（here_2）

(注意有可能CUDA安装失败，原因可能是VS版本的关系，可以尝试先卸载VS，再重新安装)

2.保证CUDA版本与Nvidia版本对应（查看here）

(注意有可能CUDA安装失败，原因可能是VS版本的关系，可以尝试先卸载VS，再重新安装)

3.*实测目前RTX3060不支持CUDA 11以下版本，会报错：cuda的算力要支持你的显卡算力*

4.下载对应版本的torch、torchvision（查看here）

torch和cudatoolkit对应版本参考（here）

下载：

在终端中进行安装（一定要在自己指定的环境，pycharm默认是当前使用的虚拟环境）

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
# 注意我使用的是cuda11.1版本，对应的是torch==1.9.0，torchvision==0.10.0
# 但是也可以是别的版本的，可以自行查看
# -f 网址就是上面的链接网址，里面可以查找到这些要下载的

5.检查是否安装正确版本后，验证是否可用

import torch
print(torch.cuda.is_available()) # 输出True说明现在CUDA可用