一定保证版本要对应
1.首先是有N卡,然后保证安装好Nvidia驱动程序后,安装CUDA
(注意有可能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可用