温馨提示×

温馨提示×

您好,登录后才能下订单哦!

密码登录×
登录注册×
其他方式登录
点击 登录注册 即表示同意《亿速云用户服务条款》

Ubuntu14.04 CuDNN6.5如何安装

发布时间:2021-11-15 17:31:38 来源:亿速云 阅读:171 作者:小新 栏目:大数据

这篇文章主要为大家展示了“Ubuntu14.04 CuDNN6.5如何安装”,内容简而易懂,条理清晰,希望能够帮助大家解决疑惑,下面让小编带领大家一起研究并学习一下“Ubuntu14.04 CuDNN6.5如何安装”这篇文章吧。

环境

系统:Ubuntu 14.04 x64
CUDA:cuda7 (安装cudnn前必须安装了cuda,具体安装见本博客相关文章http://blog.csdn.net/honyniu/article/details/46387429)

简介

CuDNN是专门针对Deep Learning框架设计的一套GPU计算加速方案,目前支持的DL库包括Caffe,ConvNet, Torch7等。在这里安装主要是为了后期caffe使用cudnn加速,当然这个在使用caffe的过程中不是必须的。
CuDNN可以在官网免费获得,注册帐号后即可下载。
下载链接:https://developer.nvidia.com/cuDNN

安装包

在这里我下载了两个版本的安装包,注册后都可以下载,如下

cudnn-6.5-linux-R1cudnn-6.5-linux-x64-v2

这两个安装包下的文件名称和结构大体是一样,不同的是cudnn的这两个版本库提供的接口完全不同(具体可以看提供的CUDNN_Library.pdf接口文档),而且caffe在早期的版本使用的是R1接口,后来最新的就变成v2接口了,完全不能通用。现在还有许多项目是基于早期caffe的,如果想使用这些项目的话,就要安装R1,同时在最新的caffe上面也有需求,所以在两个都要使用,在不同的编译情况下进行切换就可以了。

安装

基本原理是把lib文件加入到系统能找到的lib文件夹里, 把头文件加到系统能找到的include文件夹里就可以,在这里放到cuda的相应的安装目录下(前提是cuda已经安装成功了)
对R1和v2安装流程一模一样,

cd cudnn_dirsudo cp lib* /usr/local/cuda/lib64/sudo cp cudnn.h /usr/local/cuda/include/

当你想从R1切换到v2时,只需要把v2新安装一遍就可以了,同理从v2到R1

错误

如果编译caffe时开启’USE_CUDNN := 1’的话,没有上面那样把相关的文件拷入,编译时就会报错

./include/caffe/util/cudnn.hpp:5:19: fatal error: cudnn.h: 没有那个文件或目录 #include <cudnn.h>

以上是“Ubuntu14.04 CuDNN6.5如何安装”这篇文章的所有内容,感谢各位的阅读!相信大家都有了一定的了解,希望分享的内容对大家有所帮助,如果还想学习更多知识,欢迎关注亿速云行业资讯频道!

向AI问一下细节

免责声明:本站发布的内容(图片、视频和文字)以原创、转载和分享为主,文章观点不代表本网站立场,如果涉及侵权请联系站长邮箱:is@yisu.com进行举报,并提供相关证据,一经查实,将立刻删除涉嫌侵权内容。

AI