centos7安装cudnn

如题

下载

地址：https://developer.nvidia.com/rdp/cudnn-download
cudnn需要注册登录才能下载，然后选择对应的版本进行下载，最开始我下载的是Local Installer for RedHat/Centos 7.1 x84_64 (RPM)，然后使用sudo yum install命令安装这个rpm包，最后发现不知道装哪了，也找不到cudnn的头文件。最后只能再下载另一个文件Local Installer for Linux x86_64 (Tar)，即这个tar包。
下载下来是个xz格式压缩包，首先解压：

xz -xzJf *.xz

解压后会有两个目录：include 和 lib

拷贝对应的文件到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*

按照官网的方法会可能会有bug，可能现在新的cudnn版本信息没有包含在cudnn.h里，所以我这里把include下的文件都进行了拷贝

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

最后使用cmake进行编译使用libtorch的时候成功找到了cudnn。

参考：https://docs.nvidia.com/deeplearning/cudnn/archives/cudnn_764/cudnn-install/index.html#install-linux
http://zhaoxuhui.top/blog/2021/04/13/libtorch-installation-and-use.html