搭建环境(三)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

共两个压缩包(解压后145M),此为其一 环境: windows10(X64) vs2015 Anaconda3-4.4.0-Windows-x86_64(python3.6) cuda_8.0.61_win10(basic) cuda_8.0.61.2_windows(Patch 2) cudnn-8.0 tensorflow1.30 pip install tensorflow-gpu后 Anaconda prompt中输入:import tensorflow时,报错:ImportError,需要用到这个库文件,分别copy至CUDA\v8.0 中个目录下:bin,include,lib 具体错误一般如下: File "C:\Anaconda3\lib\site-packages\tensorflow\python\pywrap_tensorflow.py", line 18, in swig_import_helper return importlib.import_module(mname) File "C:\Anaconda3\lib\importlib\__init__.py", line 126, in import_module return _bootstrap._gcd_import(name[level:], package, level) File "", line 986, in _gcd_import File "", line 969, in _find_and_load File "", line 958, in _find_and_load_unlocked File "", line 666, in _load_unlocked File "", line 577, in module_from_spec File "", line 906, in create_module File "", line 222, in _call_with_frames_removed ImportError: DLL load failed: The specified module could not be found. During handling of the above exception, another exception occurred: Traceback (most recent call last): File "C:\Anaconda3\lib\site-packages\tensorflow\python\__init__.py", line 66, in from tensorflow.python import pywrap_tensorflow File "C:\Anaconda3\lib\site-packages\tensorflow\python\pywrap_tensorflow.py", line 21, in _pywrap_tensorflow = swig_import_helper() File "C:\Anaconda3\lib\site-packages\tensorflow\python\pywrap_tensorflow.py", line 20, in swig_import_helper return importlib.import_module('_pywrap_tensorflow') File "C:\Anaconda3\lib\importlib\__init__.py", line 126, in import_module return _bootstrap._gcd_import(name[level:], package, level) ImportError: No module named '_pywrap_tensorflow' During handling of the above exception, another exception occurred: 。。。。。
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值