为了使用GPU运行python程序,进行深度学习,下载NVIDIA CUDA已成功。安装CUDA的方法详见:
https://blog.csdn.net/weixin_53762670/article/details/131845364
win-R输入cmd命令,调出命令窗口,输入:
nvcc --version
如果出现以下信息即为CUDA安装成功
安装cuDNN Archive
cuDNN 其实就是 CUDA 的一个补丁而已,专为深度学习运算进行优化的
下载cuDNN Archive
在官网中下载与CUDA对应版本的cuDNN Archive
https://developer.nvidia.com/rdp/cudnn-archive
我的CUDA是12.2.0,所以下载12.0x的
这里有很多cuDNN的版本可以选,选12.0x(对于我而言)最新版本的就可以了
在下载之前,需要注册NVIDIA账号
注册结束后就可以开始下载了
复制补丁到CUDA中
下载得到压缩包,并解压,得到cudnn-windows-x86_64-8.9.2.26_cuda12-archive,内容如下所示,将这三个文件夹复制到之前安装CUDA文件夹中,覆盖掉。
(我忘记了CUDA的安装文件夹了,当时选择的是默认安装路径,可以参考这张图)
配置环境
在CUDA中找到这四个文件夹的路径,并将其添加到环境变量path中
任务栏搜索“编辑系统环境变量”–“高级”–“环境变量”–“Path”
检验安装是否成功
调出cmd命令窗口,cd命令到安装目录
cd C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v12.2\extras\demo_suite
分别执行bandwidthTest.exe和deviceQuery.exe
- 先执行bandwidthTest.exe
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v12.2\extras\demo_suite>.\bandwidthTest.exe
这里显示result为pass,说明bandwidthTest.exe运行成功
2. 下面运行deviceQuery.exe
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v12.2\extras\demo_suite>.\deviceQuery.exe
这里也能看到result为pass
至此,安装成功!
参考信息
https://blog.csdn.net/pengxiang1998/article/details/125343621