Ubuntu 16.04系统安装cuDNN 5.1流程详解

最近研究实现基于计算机视觉的目标识别技术，需要在Ubuntu系统中搭建深度学习环境，和大家分享Ubuntu 16.04系统深度学习环境搭建教程，希望能帮助到有需要的朋友。

Ubuntu 16.04系统深度学习环境搭建 Part 3——安装cuDNN 5.1

1.进入NVIDIA官网下载对应版本的cuDNN（下载需要注册NVIDIA账号并登录）。
NVIDIA官网地址：https://developer.nvidia.com/rdp/cudnn-archive

2.下载完成后，将下载文件拷贝至home文件夹下，解压缩文件。

sudo tar -zxvf ./cudnn-8.0-linux-x64-v5.1.tgz

3.拷贝头文件和库文件至usr目录对应文件夹。

sudo cp ~/cuda/include/cudnn.h /usr/local/cuda-8.0/include
sudo cp ~/cuda/lib64/lib* /usr/local/cuda-8.0/lib64

4.进入/usr/local/cuda/lib64文件夹，删除原有软件链接。

cd /usr/local/cuda/lib64
sudo rm -rf libcudnn.so libcudnn.so.5

5.建立新的软件链接。

sudo ln -s libcudnn.so.5.0.5 libcudnn.so.5
sudo ln -s libcudnn.so.5 libcudnn.so

6.使链接立即生效。

sudo ldconfig

7.验证是否安装成功。

nvcc -V

若出现相关版本信息，则表示安装成功！