cudnn是nvidia适配的深度开发套件
下面以cuDDN8.6 在ubuntu2204下的安装
参考链接
链接: cuda安装
一、安装包下载
链接: cuDNN官方下载地址
链接:cuDNN官网文档
cuDNN下载需要登录Nvidia账号,并且加入开发者(下载的时候会提示)
下载需要的版本和对应的平台,如果是windows,选择第一个zip的
二、linux安装
1.上传到服务器
进入到目录(根据自己的需求)
cd /usr/local
2.安装
dpkg -i cudnn-local-repo-ubuntu2204-8.6.0.163_1.0-1_amd64.deb
会显示需要先复制GPG Key,根据提示复制后,再次运行上面的命令
3.检查包
dpkg -l | grep cudnn
有结果就是装好了
三、windows安装
1.打开压缩包解压
2.找到cuda的目录
比如我的是
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v12.2
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
分别运行下面的程序,不报错即可
deviceQuery.exe
bandwidthTest.exe