首先安装显卡对应的CUDA版本,关键点在于区别显卡支持的CUDA最高版本和运行版本
1、查看当前显卡支持的最高版本,有两种方式:
1)NVIDIA控制面板—>帮助—>系统信息—>组件—>NVCUDA.dll对应版本
请注意,12.2为本机CUDA支持的最高版本
nvidia-smi
nvidia-smi显示的同上,也表示cuda支持的最高版本
安装对应CUDA版本并查看
按照CUDA时需要对应torch的版本,官方推荐的cuda版本为10.2和11.3,这两种 cuda 支持大多数的 pytorch 版本。
1、以下是Pytorch和CUDA对应的版本
2、以下是Pytorch和Python对应的版本
3、安装CUDA 11.3版本
1)CUDA 安装包官网
下载对应安装包并默认按照即可,安装效果如下:
上述11.3即对应的cuda版本
ncvv -V #查看并检查电脑对应cuda版本