cudnn安装

2020-11-30 本文已影响0人天空之翼

切换到压缩包所在位置，解压

$ 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