ubuntu下安装cuDNN

正确的安装cuDNN方式，其实跟着官方安装说明进行安装就可以了。

#####################################################################

1. 从https://developer.nvidia.com/cudnn上下载cudnn相应版本的压缩包（需要注册或登录）。
下载cuda对应版本的cudnn：cuDNN Library for Linux

解压当前的.tgz格式的软件包

tar -xzvf cudnn-9.0-linux-x64-v7.tgz

解压后的文件夹名为cuda，文件夹中包含两个文件夹：一个为include，另一个为lib64。
最好把这个cuda文件夹放在home目录下，方便操作。

#####################################################################

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

sudo gedit ~/.bashrc

在最后一行加入：

export LD_LIBRARY_PATH=/home/cuda/lib64:$LD_LIBRARY_PATH

注意lib64文件夹的路径，

保存更改的文件后，紧接着：

source .bashrc

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

#####################################################################

3. 配置cuDNN的最后一步就是将解压后的/home/cuda/include目录下的一些文件拷贝到/usr/local/cuda/include中。

由于进入了系统路径，因此执行该操作时需要获取管理员权限。

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

到此cuDNN的配置就完成了。

#####################################################################

猜你喜欢