【环境搭建与软件安装】How to install CUDNN or uninstall

前言

CuDnn是用于深度学习的GPU加速库，安装好NVIDIA和CUDA之后，安装CuDnn就简单多了，可参考官方文档。

操作过程

1. 下载cuDnn.

需要在NVIDIA官网注册账号，登录之后下载所需要版本的cudnn文件。一定要匹配你需要的环境和软件版本，博主下载的是*tar文件。

博主安装NVIDIA和CUDA时候都是按照官网的步骤和安装目录默认安装的。

2. 解压缩tar文件。

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

注意，这里只是一个参考例子，如果解压缩出现问题，可参考here.

3. 复制文件至CUDA目录，并更改文件权限。

$ 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 /usr/local/cuda/lib64/libcudnn*

tar文件没有验证cudnn是否正常安装，但是deb文件可以验证，可参考官网。

或者(以官网为主)

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

4. 如果有之前的残留版本最好卸载干净；

sudo rm -rf /usr/local/cuda/include/cudnn.h
sudo rm -rf /usr/local/cuda/lib64/libcudnn*

5. 查看cudnn的版本；

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

注意，可能以上脚本不能直接查看cudnn版本，需要查看cudnn_version.h文件获取；

参考:

1. nvidia_install_cudnn_linux;

完