大家好,又见面了,我是你们的朋友全栈君。
一、cudnn简介
NVIDIA cuDNN是用于深度神经网络的GPU加速库。它强调性能、易用性和低内存开销。NVIDIA cuDNN可以集成到更高级别的机器学习框架中,如谷歌的Tensorflow、加州大学伯克利分校的流行caffe软件。简单的插入式设计可以让开发人员专注于设计和实现神经网络模型,而不是简单调整性能,同时还可以在GPU上实现高性能现代并行计算。安装cudnn的前提条件是已经安装对应的cuda版本,系统环境如下:
- 操作系统:Ubuntu 18.04
- cuda版本:cuda11.4
- cudnn版本:cudnn8.2.4
二、安装步骤
1、登录官网选择cudnn版本
登录英伟达官网https://developer.nvidia.com/zh-cn/cudnn
2、注册一个英伟达网站账号
3、填写一份调查问卷
4、下载cudnn 11.4
5、将下载的deb软件包上传到服务器
root@testuat:/usr/local/cuda# ll |grep cudnn8
-rw-r–r– 1 ubuntu ubuntu 512181782 9月 3 17:10 libcudnn8_8.2.4.15-1+cuda11.4_amd64.deb
-rw-r–r– 1 ubuntu ubuntu 381367990 9月 3 17:10 libcudnn8-dev_8.2.4.15-1+cuda11.4_amd64.deb
-rw-r–r– 1 ubuntu ubuntu 1669692 9月 3 17:27 libcudnn8-samples_8.2.4.15-1+cuda11.4_amd64.deb
6、安装libcudnn8
root@testuat:/usr/local/cuda# sudo dpkg -i libcudnn8
libcudnn8_8.2.4.15-1+cuda11.4_amd64.deb libcudnn8-dev_8.2.4.15-1+cuda11.4_amd64.deb
root@testuat:/usr/local/cuda# sudo dpkg -i libcudnn8_8.2.4.15-1+cuda11.4_amd64.deb
正在选中未选择的软件包 libcudnn8。
(正在读取数据库 … 系统当前共安装有 222327 个文件和目录。)
正准备解包 libcudnn8_8.2.4.15-1+cuda11.4_amd64.deb …
正在解包 libcudnn8 (8.2.4.15-1+cuda11.4) …
正在设置 libcudnn8 (8.2.4.15-1+cuda11.4) …
正在处理用于 libc-bin (2.27-3ubuntu1.4) 的触发器 …
7、安装libcudnn8-dev
root@testuat:/usr/local/cuda# sudo dpkg -i libcudnn8-dev_8.2.4.15-1+cuda11.4_amd64.deb
(正在读取数据库 … 系统当前共安装有 222345 个文件和目录。)
正准备解包 libcudnn8-dev_8.2.4.15-1+cuda11.4_amd64.deb …
正在将 libcudnn8-dev (8.2.4.15-1+cuda11.4) 解包到 (8.2.4.15-1+cuda11.4) 上 …
正在设置 libcudnn8-dev (8.2.4.15-1+cuda11.4) …
update-alternatives: 使用 /usr/include/x86_64-linux-gnu/cudnn_v8.h 来在自动模式中提供 /usr/include/cudnn.h (libcudnn)
8、安装libcudnn8-samples
root@testuat:/usr/local/cuda# sudo dpkg -i libcudnn8-samples_8.2.4.15-1+cuda11.4_amd64.deb
正在选中未选择的软件包 libcudnn8-samples。
(正在读取数据库 … 系统当前共安装有 222345 个文件和目录。)
正准备解包 libcudnn8-samples_8.2.4.15-1+cuda11.4_amd64.deb …
正在解包 libcudnn8-samples (8.2.4.15-1+cuda11.4) …
正在设置 libcudnn8-samples (8.2.4.15-1+cuda11.4) …
9、拷贝cudnn_samples_v8到用户家目录
(base) ubuntu@testuat:~$ cp -r /usr/src/cudnn_samples_v8/ $HOME
10、编译mnistCUDNN
(base) ubuntu@testuat:~$ cd $HOME/cudnn_samples_v8/mnistCUDNN
(base) vanfon@vanfonuat:~/cudnn_samples_v8/mnistCUDNN$ make clean && make
11、检查验证
(base) ubuntu@testuat:~/cudnn_samples_v8/mnistCUDNN$ ./mnistCUDNN
三、QA
1、编译mnistCUDNN是报错
- 报错信息
- 解决方案
(base) ubuntu@testuat:~/cudnn_samples_v8/mnistCUDNN$ sudo apt-get install libfreeimage3 libfreeimage-dev
发布者:全栈程序员-用户IM,转载请注明出处:https://javaforall.cn/146024.html原文链接:https://javaforall.cn
【正版授权,激活自己账号】: Jetbrains全家桶Ide使用,1年售后保障,每天仅需1毛
【官方授权 正版激活】: 官方授权 正版激活 支持Jetbrains家族下所有IDE 使用个人JB账号...