安装完cuda之后 安装cudnn

nvcc --version（自己安装的cuda版本） 版本需要和nvidia-smi查看到的cuda版本（nvidia驱动对应的cuda版本）一致

1、解压：tar zxvf cudnn-10.1-linux-x64-v7.5.0.56.tgz


include lib64两个文件夹

２、cuda安装目录为/usr/loca/cuda-10.1

把cudnn的include/ 和 lib64/目录下的文件拷贝到cuda的安装目录下的include/ 和 lib64里面

$ sudo cp -P include/cudnn.h /usr/local/cuda-10.1/include
$ sudo cp -P lib64/libcudnn* /usr/local/cuda-10.1/lib64
$ sudo chmod a+r /usr/local/cuda-10.1/lib64/libcudnn*