1、查看NVIDIA显卡驱动支持的CUDA版本
WIN+R并输入CMD打开命令提示符,输入nvidia-smi查看驱动支持的最高CUDA版本,以本人驱动为例,适用的最高CUDA版本为12.6。
nvidia-smi
2、CUDA下载安装
CUDA下载网址:CUDA Toolkit Archive | NVIDIA Developer
选择需要下载的CUDA版本后,根据电脑配选择对应的选项下载(建议下载向下兼容的CUDA版本,不要选择最高兼容的CUDA版本,本人选择的CUDA版本为12.3.2)。下载好后点击安装,安装到默认路径,所有选项选择默认选项即可。
安装结束后,WIN+R并输入CMD打开命令提示符,在命令提示符里面输入nvcc -V即可以看到安装的Cuda版本。
nvcc -V
3、cuDNN下载安装
介绍:cuDNN提供了高效的算法和优化技巧,可以极大地提升深度学习模型的训练和推理速度。
cuDNN下载网址:cuDNN 历史版本 | NVIDIA 开发者
根据下载的CUDA版本下载对应的cuDNN
下载完成后解压压缩包,打开解压后的文件夹如下,同时打开C盘中存放CUDA的文件夹C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v12.3。打开后将cuDNN文件夹中bin、include、lib的文件复制到C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v12.3文件夹中对应名称的文件夹中,通过管理员权限进行替换。
未完待续~