根据安装的cuda版本下载对应的cudnncudnn下载地址
下载需要注册NVIDIA账号
解压下载好的文件(.tgz)
tar -zxvf cudnn-*******.tgz
进入解压的文件夹,进入 include 路径下,执行:
$ sudo cp cudnn.h /usr/local/cuda/include/
lib64 路径下,执行:
$ sudo cp lib* /usr/local/cuda/lib64/ //复制动态链接库
$ cd /usr/local/cuda/lib64/
$ sudo rm -rf libcudnn.so libcudnn.so.^ //^内容根据cudnn版本填写如 .7 或.8
$ sudo ln -s libcudnn.so.*.*.* libcudnn.so.^ //so.*.*.* 填写cudnn的具体版本
$ sudo ln -s libcudnn.so.^ libcudnn.so
安装完成后
$ nvcc -V 验证是否安装成功