caffe配置cuDNN

来源：互联网发布：linux查看指定目录权限编辑：程序博客网时间：2024/05/16 05:40

下述教程均基于Ubuntu14.04，且默认已安装好caffe所需要的所有依赖库以及cuda。如未安装，请先找教程安装好。

1. 首先从Github上下载caffe。

2. 解压缩，并创建Makefile.config文件。

unzip caffe-master.zip

cd caffe-master

cpMakefile.config.exampleMakefile.config

3.编辑Makefile.config，取消注释USE_CUDNN := 1

4. https://developer.nvidia.com/cudnn 注册账号（可能需要一晚上的验证）下载最新（v2以上）版的cuDNN。

5. 安装cuDNN(以R1版本为例，注意换为自己下载的包名)

tar -xzvf cudnn-6.5-linux-R1.tgz

    cd cudnn-6.5-linux-R1    sudo cp lib* /usr/local/cuda/lib64/    sudo cp cudnn.h /usr/local/cuda/include/

    cd /usr/local/cuda/lib64/

    sudo rm -rf libcudnn.so libcudnn.so.6.5

    
    sudo chmod u=rwx,g=rx,o=rx libcudnn.so.6.5.18     sudo ln -s libcudnn.so.6.5.18 libcudnn.so.6.5    sudo ln -s libcudnn.so.6.5 libcudnn.so
6.build

    make all

    make test

    make runtest

0 0