首先要看电脑支持的最高版本是多少
1、从NVIDA控制面板查看。
在左面右键点击NVIDA控制面板
点击系统信息
点击组件就能看到支持的最高版本了,我的最高版本是11.7
2、使用nvidia-smi命令查看
以我的为例,我支持最高版本为11.7,但是我打算安装11.5的cuda。
#cuda 版本安装网址
https://developer.nvidia.com/cuda-toolkit-archive
#cuDNN 版本安装网址
https://developer.nvidia.com/rdp/cudnn-archive
需要注意两个安装版本一定要一样,或者兼容。
以我的为例子,我cuda 安装11.5.2版本
我 cuDNN安装的是对应11.x的v8.3.3版本。cuDNN v8.3.3与CUDA Toolkit 11.5.2是兼容的,同时,CUDA Toolkit 11.5.2也是一个很好的版本,具有很好的稳定性和功能性。使用cuDNN v8.3.3和CUDA Toolkit 11.5.2的组合可以提高的机器学习应用的训练速度和性能,因此也建议使用这个组合。
下载cuda与cuDNN
后面的安装步骤可以观看这位B站博主:【CUDA安装/多CUDA兼容】Windows深度学习环境配置_哔哩哔哩_bilibili