1. 查看本机的CUDA驱动适配版本
桌面右键打开英伟达控制面板,点击帮助->系统信息->组件
可以看到本机支持的是CUDA11.1版本,表示的是不支持更高版本的。如果你升级了驱动,可能会支持更高版本,也可能不会提升。所以就直接安装CUDA11.1就好啦。
2. 下载和安装CUDA
2.1. 下载CUDA
下载地址:https://developer.nvidia.com/zh-cn/cuda-downloads
选择好操作系统、版本、安装程序类型就可以直接点击下载了。
2.2. 安装CUDA
下载完之后,找到下载好的CUDA,如下图,双击安装。
如下图,直接OK就行了。
选择同意并继续。
如下图,建议选择自定义安装。
如下图,我直接默认安装了,如果你想自定义安装位置的话要记住你选择的安装路径,
如下图,直接勾选上就好了,安装就可以了。
2.3. 测试是否安装成功
直接终端输入命令nvcc --version
,即可显示
版本号,如下图:
使用命令set cuda
,可以查看cuda设置的环境变量。
3. 下载和安装cuDNN
3.1. 下载cuDNN
下载地址:https://developer.nvidia.com/rdp/cudnn-download
下载cuDNN是需要登录英伟达开发者账户的,如果没有的话就注册一下并填写个问卷就行啦。
选择合适的版本下载就行啦,注意:一定要选择和你安装的CUDA匹配的版本。
3.2. 安装cuDNN
下载好之后,直接解压cuDNN压缩包,可以看到bin、include、lib目录,如下图:
打开C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA,找到bin、include、lib目录,如下图:
接下来将cuDNN压缩包内对应的bin、include、lib目录下的文件对应的复制到bin、include、lib目录下。
● 注意:是复制文件夹下的文件到bin、include、lib目录,不是复制目录。
4. 添加环境变量
5. 检查安装结果
打开C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.2\extras\demo_suite,在此路径下打开powershell(shift+鼠标右键选择在此处打开Powershell窗口(S)),执行命令.\bandwidthTest.exe
,显示结果如下图:
OK,完事儿~