jetson TX2上如何安装cudnn

https://blog.csdn.net/lucifer_zzq/article/details/76675239  此为博客来源 谢谢大神。以下全为链接。(自己写的红色标出)  

记下来完全是因为自己是小白  搞了好久  难受

 

 

首先说一下网上大多数中文安装cuDNN教程的错误方式,这种方式真的坑人无数。

简单地说网上的大多错误的安装cuDNN的方式都是将下载后的cuDNN压缩包解压。然后再将cudnn的头文件(cuda/include目录下的.h文件)复制到cuda安装路径的include路径下,将cudnn的库文件(cuda/lib64目录下的所有文件)复制到cuda安装路径的lib64路径下。这种方法如果不重置cuDNN相应的符号链接的话是不能成功的安装cuDNN的。(这个Windows下面可以用

 

下面我们说一下正确的安装cuDNN方式,其实跟着官方安装说明进行安装就可以了。(Linux下,亲测是在jetson tx2开发板上  Ubuntu16.04

从https://developer.nvidia.com/cudnn上下载cudnn相应版本的压缩包(可能需要注册或登录)。

如果这个压缩包不是.tgz格式的,把这个压缩包重命名为.tgz格式。解压当前的.tgz格式的软件包到系统中的任意路径(这个路径很重要,以下将该路径的绝对路径简称为/your/path/to/cudnn),解压后的文件夹名为cuda,文件夹中包含两个文件夹:一个为include,另一个为lib64。
例如:我将这个压缩包解压在了/usr/local目录下,那么该文件的绝对路径为/usr/local/cuda

tar -zvxf archive_name.tar.gz -C /usr/local


将解压后的文件中的lib64文件夹关联到环境变量中。这一步很重要。
   

cd ~
sudo gedit .bashrc


在弹出的gedit文档编辑器(.bashrc中)中最后一行加入:

 

export LD_LIBRARY_PATH=/your/path/to/cuda/lib64:$LD_LIBRARY_PATH



其中/your/path/to/cudnn/lib64是指.tgz解压后的文件所在路径中的lib64文件夹。
保存更改的文件后,紧接着:

 

source .bashrc



再重启一下Terminal(终端),该步骤可以成功的配置cuDNN的Lib文件。

配置cuDNN的最后一步就是将解压后的cuDNN文件夹(一般该文件名为cuda)中的include文件夹(/your/path/to/cudnn/include)中的cudnn.h文件拷贝到/usr/local/cuda/include中,由于进入了系统路径,因此执行该操作时需要获取管理员权限。

打开终端,进入/your/path/to/cudnn/include。其中/your/path/to/cudnn/include指的是.tgz解压后的文件所在路径中的include文件夹。例如:

  

cd /YOUR/PATH/TO/cuda/include
sudo cp *.h /usr/local/cuda/include/


其中这里的cuda/include对于我自己的安装来说就是/your/path/to/cudnn/include。因为我将cudnn的.tgz压缩包解压到了home的当前用户的路径下,解压后的文件夹名为cuda。

之后,再重置cudnn.h文件的读写权限:

sudo chmod a+r /usr/local/cuda/include/cudnn.h



至此,cuDNN的配置就全部安装完成了。

另外:
查看cuda版本的方法:

cat /usr/local/cuda/version.txt


查看cudnn版本的方法:

cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2


--------------------- 
作者:每天都要深度学习 
来源:CSDN 
原文:https://blog.csdn.net/lucifer_zzq/article/details/76675239 
版权声明:本文为博主原创文章,转载请附上博文链接!

你可能感兴趣的:(jetson TX2上如何安装cudnn)