安装CUDA10.2对应的cuDNN
1、进入官网,根据CUDA版本选择cuDNN版本
https://developer.nvidia.com/rdp/cudnn-download
2、没有账号的需要注册账号,登录
3、选择cuDNN Library for Linux即可,如下图所示。
4、解压cuDNN包.
tar -xzvf cudnn-x.x-linux-x64-v8.x.x.x.tgz
or
tar -xzvf cudnn-x.x-linux-aarch64sbsa-v8.x.x.x.tgz
5、下载解压之后,将cuda/include/cudnn.h文件复制到usr/local/cuda/include文件夹,将cuda/lib64/下所有文件复制到/usr/local/cuda/lib64文件夹中,并添加读取权限。
$ 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*
6、安装完成。