首页 > 系统服务 > 详细

ubuntu16.04安装cuDNN

时间:2019-09-15 13:05:53      阅读:111      评论:0      收藏:0      [点我收藏+]

cudnn的安装非常简单

(1)下载安装文件

按需求下载cudnn的安装文件https://developer.nvidia.com/rdp/cudnn-archive

Tar File的下载如下图所示,选择红方框中的选项进行下载

 

技术分享图片

下载的是cudnn-*tgz的压缩包时,按下方指令进行安装:

首先解压缩下的cudnn压缩包文件

tar -xzvf cudnn-9.0-linux-x64-v7.tgz

执行安装,其实就是拷贝头文件和库文件并给予权限

sudo cp cuda/include/cudnn.h /usr/local/cuda/include

sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64

sudo chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda/lib64/libcudnn*

(2)验证安装是否成功

cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2

如果显示

#define CUDNN_MAJOR 7
#define CUDNN_MINOR 6
#define CUDNN_PATCHLEVEL 2
--
#define CUDNN_VERSION (CUDNN_MAJOR * 1000 + CUDNN_MINOR * 100 + CUDNN_PATCHLEVEL)

则cudnn安装成功

ubuntu16.04安装cuDNN

原文:https://www.cnblogs.com/liuwenhua/p/11521668.html

(2)
(2)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!