tips:cudnn要与cuda对应版本下载
1.下载对应cuda版本的cudnn,需要注册Nvidia开发者账号。NVIDIA cuDNN下载地址:https://developer.nvidia.com/cudnn
2.解压:tar zxvf cudnn*.tgz
3.
cuda的安装目录通过which nvcc
查看
cuda安装目录为/usr/loca/
cuda-10.2
4.把include/ 和 lib64/
目录下的文件拷贝到cuda的安装目录下的include/ 和 lib64
里面
sudo cp -P include/cudnn.h /usr/local/cuda-10.2/include
sudo cp -P lib64/libcudnn* /usr/local/cuda-10.2/lib64
sudo chmod a+r /usr/local/cuda-10.2/lib64/libcudnn*