搭建环境(三)Ubuntu14.04 安装cudnn-v6.0

1.介绍

NVIDIA cuDNN是用于深度神经网络的GPU加速库。它强调性能、易用性和低内存开销。NVIDIA cuDNN可以集成到更高级别的机器学习框架中。简单的,插入式设计可以让开发人员专注于设计和实现神经网络模型,而不是调整性能,同时还可以在GPU上实现高性能现代并行计算。cuDNN为深度神经网络中的标准流程提供了高度优化的实现方式,例如convolution、pooling、normalization以及activation layers的前向以及后向过程。目前的深度学习框架都支持cuDNN:Caffe、Caffe2、TensorFlow、Torch、Pytorch、Theano等。
官方文档 https://pan.baidu.com/s/1dFjclyD

2.下载cudnn资源

 官网地址地址<https://developer.nvidia.com/cudnn>(需要注册或登录)
 我CUDA装的是8.0版本,然后cudnn装了5.1。这个版本选择和你安装的TensorFlow版本也有关系,我后面先安装了TensorFlow0.5.0,运行是没有问题的。但是升级成了TensorFlow1.3.0的时候就报错了,这个版本要求cudnn版本至少是v6.0以上,大多数的教程都写的比较早,多没有讲到,所以为了后边少出问题,建议安装v6.0(或者更高版本,官网已经出到cudnn7了)

3.安装

说是安装,其实就是解压。把cudnn压缩文件拷贝到CUDA目录下(其实其他目录也可以)

tar -xzvf cudnn-6.0-linux-R1.tgz
#解压后会出来一个CUDA的文件夹,网上教程都是cudnn-X.0的文件夹,解压了几次都是,也不知道什么原因,不过不重要。
cd cuda
#在lib64目录下复制文件:
sudo cp lib* /usr/local/cuda-8.0/lib64/
#在include目录下复制文件:
sudo cp cudnn.h /usr/local/cuda-8.0/include/

再cd进入lib64目录下的动态文件进行复制和链接:
(这一步是必须的,不然运行程序的时候会出现:“error while loading shared libraries: libcudnn.so.x.x: cannot open shared object file: No such file or directory“的错误,同理,如果你后期出现了这样的报错,肯定就是这里的问题)
6.0.8为对应版本具体可修改

cd /usr/local/cuda/lib64/
#删除原有动态文件
sudo rm -rf libcudnn.so libcudnn.so.5 
#生成软衔接
sudo ln -s libcudnn.so.6.0.8 libcudnn.so.5 
#生成软链接
sudo ln -s libcudnn.so.5 libcudnn.so 

至此,cudnn就算是装好了,如果还有问题的话,在网上看到有一篇教程写到需要设置环境变量,可以试一下能否解决,不过我没有用到。
链接 http://blog.csdn.net/lucifer_zzq/article/details/76675239

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值