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-R1
cudnn-6.5-linux-x64-v2

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

安装

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

cd cudnn_dir
sudo 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>

参考

http://www.cnblogs.com/platero/p/4118139.html (如果执行上面的操作还是找不到cudnn的库,可以参考这个文章,没遇到这个问题,在这里就不写了)

  • 0
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值