问小白 wenxiaobai
资讯
历史
科技
环境与自然
成长
游戏
财经
文学与艺术
美食
健康
家居
文化
情感
汽车
三农
军事
旅行
运动
教育
生活
星座命理

CUDA与CUDNN在Windows下的安装与配置

创作时间:
作者:
@小白创作中心

CUDA与CUDNN在Windows下的安装与配置

引用
CSDN
1.
https://blog.csdn.net/weixin_42589048/article/details/146275362

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安装成功了。

© 2023 北京元石科技有限公司 ◎ 京公网安备 11010802042949号