cudnn安装

https://developer.nvidia.com/rdp/cudnn-download
https://docs.nvidia.com/deeplearning/sdk/cudnn-install/index.html#installlinux-tar

下载

之前下载和安装的驱动和cuda版本如下

+-----------------------------------------------------------------------------+
| NVIDIA-SMI 418.43       Driver Version: 418.43       CUDA Version: 10.1     |
|-------------------------------+----------------------+----------------------+

所以我们只需要安装对应的cudnn即可

重命名格式

cp cudnn-10.1-linux-x64-v7.5.0.56.solitairetheme8 cudnn-10.1-linux-x64-v7.5.0.56.tgz

解压

tar -xzvf cudnn-10.1-linux-x64-v7.5.0.56.tgz
#解压之后会生成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*