Cudnn安装详细步骤

来源:互联网 发布:淘宝怎么刷单提升权重 编辑:程序博客网 时间:2024/06/05 09:20

CuDNN是专门针对Deep Learning框架设计的一套GPU计算加速方案,目前支持的DL库包括Caffe,ConvNet, Torch7等。

CuDNN可以在官网免费获得,注册帐号后即可下载。官网没有找到安装说明,下载得到的压缩包内也没有Readme. 不过google一下就会找到许多说明。基本原理是把lib文件加入到系统能找到的lib文件夹里, 把头文件加到系统能找到的include文件夹里就可以。

官网上的教程

This guide provides step-by-step instructions on how to install and check for correct operation of NVIDIA cuDNN v7.0.5 on Linux, Mac OS X, and Microsoft Windows systems.http://docs.nvidia.com/deeplearning/sdk/cudnn-install/index.html 


cudnn安装注意点: 

cudnn的安装其实很简单,关键点是一定要安装cuda对应的cudnn包,本机中安装的cuda7.5所以对应的cudnn为v5.1这很重要,我就是安装错了版本,导致后面caffe的编译总是出错。 
cudnn安装步骤: 
1、从官网上下载cudnn的安装包。 
2、将安装包解压,将此安装包放在home路径下即可,并在当前路径下进行解压,解压后的文件夹名为cuda。 (注意文件名)

tar -xzvf cudnn-6.5-linux-R1.tgz

3、在终端上编辑如下代码:

$ cd cuda/include $ sudo cp *.h /usr/local/cuda/include/  (注意这里是你自己的cuda的安装地址,复制*.h文件) 
  • 1
  • 2
  • 3

4、关闭终端并重新打开(也可以返回最开始的目录)

$ cd cuda/lib64 $ sudo cp libcudnn* /usr/local/cuda/lib64/ 
  • 1
  • 2
  • 3

5、终端运行:

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