cudnn是nvidia适配的深度开发套件

下面以cuDDN8.6 在ubuntu2204下的安装


如果没有安装cuda,可以参考这篇

链接: cuda安装


安装包下载

链接: cuDNN官方下载地址(https://developer.nvidia.com/rdp/cudnn-archive)

cuDNN下载需要登录Nvidia账号,并且加入开发者(下载的时候会提示)

下载需要的版本和对应的平台,如果是windows,选择第一个zip的

image.png


一、linux安装

1.上传到服务器

进入到目录(根据自己的需求)


cd /usr/local/my_cuda

1

上传


2.安装

dpkg -i cudnn-local-repo-ubuntu2204-8.6.0.163_1.0-1_amd64.deb

1

会显示需要先复制GPG Key,根据提示复制后,再次运行上面的命令


3.检查包

dpkg -l | grep cudnn

1

二、windows安装

1.打开压缩包解压

2.找到cuda的目录

比如我的是


C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v12.2

1

3.将解压后的文件直接复制到cuda目录

有三个文件夹bin、include、lib,cuda安装目录下也有这三个,直接复制进去就行,文件夹会合并


4.如何验证

使用cuda自带的小工具,使用cmd运行,注意替换成你的cuda安装地址,我这里是12.2的版本


cd C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v12.2\extras\demo_suite

1

分别运行下面的程序,不报错即可


deviceQuery.exe

1

bandwidthTest.exe

————————————————

版权声明:本文为CSDN博主「ziqibit」的原创文章,遵循CC 4.0 BY-SA版权协议,转载请附上原文出处链接及本声明。

原文链接:https://blog.csdn.net/ziqibit/article/details/129960088