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
Logo

魔乐社区(Modelers.cn) 是一个中立、公益的人工智能社区,提供人工智能工具、模型、数据的托管、展示与应用协同服务,为人工智能开发及爱好者搭建开放的学习交流平台。社区通过理事会方式运作,由全产业链共同建设、共同运营、共同享有,推动国产AI生态繁荣发展。

更多推荐