安装完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两个文件夹

2、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*
posted @ 2020-02-24 10:01  U_C  阅读(712)  评论(1编辑  收藏  举报