ubuntu14.04配置caffe

ubuntu 14.04 64bit 安装

请自行Google安装,并修改源。

cuda 7.5 安装

  1. cuda7.5下载,选择ubuntu 14.04的deb包下载

  2. 安装一些可能的依赖

    sudo apt-get install linux-headers-$(uname -r) build-essential

  3. 安装cuda

    sudo dpkg –i cuda-repo-<distro>_<version>_<architecture>.deb

    sudo apt-get update

    sudo apt-get install cuda

    注: cuda的安装,并不需要先手动安装NVIDIA驱动,安装cuda会依赖NVIDIA驱动,请保持联网,会自行下载安装。

  4. 环境变量设置
    在/etc/profile文件加入

    PATH=/usr/local/cuda-7.5/bin:$PATH

    export PATH

    然后输入source /etc/profile使更改生效

  5. 安装cuda_samples

    cuda-install-samples-7.5.sh <dir>

    cd ~/NVIDIA_CUDA-7.5_Samples

    make -j8 编译

    切换到bin目录,执行./deviceQuery,显示Result=PASS表示安装成功

  6. cudnn加速安装

    cudnn下载

    sudo cp cudnn.h /usr/local/cuda/include/

    sudo cp lib* /usr/local/cuda/lib64

    cd /usr/local/cuda/lib64

    sudo rm -rf libcudnn.so.7.0 libcudnn.so

    sudo ln -s libcudnn.so.7.0.64 libcudnn.so.7.0

    sudo ln -s libcudnn.so.7.0 libcudnn.so

    更改文件权限

  7. 添加lib路径

    sudo vim /etc/ld.so.conf.d/cuda.conf

    输入 /usr/local/cuda/lib64

    保存,并执行sudo ldconfig

caffe 编译

  1. 安装依赖

    sudo apt-get install -y libblas-dev liblapack-dev libopenblas-base libopenblas-dev

    sudo apt-get install -y libboost1.55-all-dev

    sudo apt-get install -y libprotobuf-dev libleveldb-dev libsnappy-dev libopencv-dev libhdf5-serial-dev

    sudo apt-get install -y libgflags-dev libgoogle-glog-dev liblmdb-dev protobuf-compiler

  2. 修改Makefile.config

    cp Makefile.config.example Makefile.config

    修改如下

    1. 取消# USE_CUDNN := 1注释
    2. 修改PYTHON_INCLUDE路径
    3. 修改MATLAB_DIR路径
    4. 取消# WITH_PYTHON_LAYER := 1注释
    5. 修改BLAS := open
  3. 编译并测试

    make all -j8

    make test -j8

    make runtest -j8

Python caffe 接口编译

  1. 安装python模块

    sudo apt-get install python-dev python-pip gfortran

    for req in $(cat requirements.txt); do sudo pip install $req; done

    如果有模块安装不成功可以多执行几次。

    pip可以修改源,加快下载速度,国内有豆瓣、v2ex等python源。使用方法,自行google。

  2. 编译pycaffe

    make pycaffe -j8

  3. ~/.bashrc中添加PYTHONPATH变量

    export PYTHONPATH=$PYTHONPATH:/home/whq/deep-learning/caffe-master/python

  4. 执行make pytest -j8测试

Matlab caffe 接口编译

  1. 安装MATLAB
  2. 降低gcc版本(MATLAB 2014a,只支持gcc4.7)或 修改Matlab mex使用的gcc版本
  3. 编译matcaffe

    make matcaffe -j8

  4. 执行make mattest -j8测试

机器硬件配置为i7 4790x、 32GB RAM、GeForce GTX TITAN,因软件硬件环境不同可能造成安装失败。

如有问题,欢迎留言指正。

posted @ 2016-04-06 13:53  吴汉卿  阅读(357)  评论(0)    收藏  举报