切换到压缩包所在位置，解压
$ tar -xzvf cudnn-8.0-linux-x64-v5.1.tgz
复制相关文件到cuda特定目录下(我的cuda安装目录为/usr/local/cuda/)
$ sudo cp cuda/include/cudnn.h /usr/local/cuda/include
$ sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64
修改文件权限
$ sudo chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda/lib64/libcudnn*
使用
这个时候编译caffe可以将Makefile.config中的USE_CUDNN不注释，从而使用cuDNN加速
# cuDNN acceleration switch (uncomment to build with cuDNN).
USE_CUDNN := 1
截取自
https://blog.csdn.net/yaoqiwaimai/article/details/78594154
网友评论