实验室服务器安装了cuda之后,需要安装cudnn加速,安装记录如下:
- 下载对应cuda版本的cudnn,需要注册Nvidia开发者账号。NVIDIA cuDNN下载地址
- 解压:
tar zxvf cudnn-8.0-linux-x64-v5.1.tgz
看到如下目录:
3.cuda的安装目录通过which nvcc
查看
cuda安装目录为
/usr/loca/cuda-8.0
4 把include/ 和 lib64/
目录下的文件拷贝到cuda的安装目录下的include/ 和 lib64
里面
$ cd ~/src/cuda #解压后的路径
$ sudo cp -P include/cudnn.h /usr/local/cuda-8.0/include
$ sudo cp -P lib64/libcudnn* /usr/local/cuda-8.0/lib64
$ sudo chmod a+r /usr/local/cuda-8.0/lib64/libcudnn*