CUDA与CUDNN在Windows下的安装与配置
CUDA与CUDNN在Windows下的安装与配置
CUDA(Compute Unified Device Architecture)和CUDNN(CUDA Deep Neural Network library)是深度学习开发中常用的工具,正确安装和配置这些工具对于使用NVIDIA显卡的开发者来说非常重要。本文将详细介绍如何在Windows系统下安装和配置CUDA和CUDNN。
一、查看电脑CUDA版本
按住win+r
,输入cmd
点击确定,进入终端窗口。
输入nvidia-smi
,查看版本号。
二、下载CUDA
我的CUDA Version为12.5,说明该显卡最高支持到12.5,我这里就选择12.4的版本,你也可以选择更低的版本。
下载网址:CUDA Toolkit Archive | NVIDIA Developer
点击后等待一会即可进行下载,我是Windows11操作系统所以选择Windows,Version选择11,选择本地下载exe(local),最后Download进行本地下载。
三、CUDA安装
使用默认路径安装,安装的时候一定要把杀毒软件关闭,否则安装失败。
同意继续
选择自定义,下一步。
把CUDA选一下,还需要注意一点,如果你前面没有安装vs,需要把CUDA里面的Visual Studio Integration取消勾选,否则会安装不成功。
默认位置,下一步。
四、检查CUDA是否安装成功
安装成功后,同时点击win+r
,输入cmd
进入终端窗口输入nvcc -V
,若结果如图所示则证明CUDA安装成功
五、CUDNN的下载
cuDNN Archive | NVIDIA Developer
这里以cuDNN for CUDA 12.x支持包为例,如果CUDA版本不是12.4的用户,下载相应的cuDNN包。点击下载后,需要你进行注册。
解压后把这三个文件夹复制到Cuda安装路径下
六、编辑系统环境变量
打开“编辑系统环境变量”,点击“环境变量”
添加以下四个路径
七、检查CUDNN是否安装成功
打开C:\Program Files\NVIDIA Computing Toolkit\CUDA\v12.4\extras\demo_suite
,查看是否有文件bandwidthTest.exe
以及deviceQuery.exe
,若存在则在该文件中打开cmd运行两.exe文件
若都能成功运行就恭喜你CUDNN安装成功了。
