ubunt16.04下caffe环境安装

一. 系统初始环境

系统:Ubuntu16.04: ubuntu-16.04-desktop-amd64.iso

cuda安装文件: cuda-repo-ubuntu1604-8-0-local_8.0.44-1_amd64.deb.44-1_amd64-deb,下载链接点击, linux-x86架构-ubuntu-16.04-deb(local)

cudnn安装文件: cudnn-8.0-linux-x64-v5.0-ga.solitairetheme8,下载链接点击, 适用cuda8.0有5.1和5.0版,这里用5.0版,区别应该不大

caffe源代码: github链接 ,或者运行git clone https://github.com/BVLC/caffe.git

二. 安装cuda8.0

1. sudo apt-get update
2. sudo apt-get upgrade
3. 删除自带显卡驱动 sudo apt-get –purge remove nvidia-*
4. sudo dpkg -i cuda-repo-ubuntu1604-8-0-local_8.0.44-1_amd64.deb.44-1_amd64-deb (这个文件包含了nvidia-367的驱动)
5. 安装cuda: sudo apt-get install cuda
6. reboot

测试安装结果:
1. cd /usr/local/cuda/samples/1_Utilities/deviceQuery
2. sudo make
3. ./deviceQuery

测试成功显示结果(部分):

deviceQuery, CUDA Driver = CUDART, CUDA Driver Version = 8.0,

CUDA Runtime Version = 8.0, NumDevs = 1, Device0 = GeForce GTX 960M

Result = PASS

如果出现Result = Failed之类的表示cuda安装失败

添加环境变量:

在/etc/profile中添加

export PATH=/usr/local/cuda-8.0/bin:$PATH export LD_LIBRARY_PATH=/usr/local/cuda-8.0/lib64:$LD_LIBRARY_PATH

三. 安装cudnn5.0

1. 解压安装包
   
   tar zxvf cudnn-8.0-linux-x64-v5.0-ga.solitairetheme8
   
2. 复制文件
   
   sudo cp cuda/include/cudnn.h /usr/local/cuda/include/
   
   sudo cp cuda/lib64/libcudnn.so.5.0.5 /usr/local/cuda/lib64/
   
3. 建立软链接
   
   cd /usr/local/cuda/lib64/
   
   sudo ln -s libcudnn.so.5.0.5 libcudnn.so.5
   
   sudo ln -s libcudnn.so.5 libcudnn.so
   

四. 安装caffe

1. 安装依赖库
   
   sudo apt-get install build-essential
   
   sudo apt-get install git cmake gedit doxygen
   
   sudo apt-get install python-numpy python-pip cython easydict
   
   sudo apt-get install libprotobuf-dev libleveldb-dev libsnappy-dev libhdf5-serial-dev protobuf-compiler
   
   sudo apt-get install libopencv-dev (想使用opencv3的可以尝试跳过这一步自行安装opencv3)
   
   sudo apt-get install –no-install-recommends libboost-all-dev
   
   sudo apt-get install libatlas-base-dev
   
   sudo apt-get install libgflags-dev libgoogle-glog-dev liblmdb-dev
   
2. 下载源代码并解压
   
   git clone https://github.com/BVLC/caffe.git
   
   unzip caffe-master.zip
   

3. 修改配置文件Make.config
   
   cd caffe-master
   
   cp Makefile.config.example Makefile.config
   
   sudo gedit Makefile.config

   开启GPU模式:将#USE_CUDNN := 1前的#注释去掉，表示使用cudnn，如果不是使用GPU，可以将#CPU_ONLY := 1前得注释去掉
   pycaffe模块配置:修改\#Whatever else you find you need goes here.
   INCLUDE_DIRS:= $(PYTHON_INCLUDE) /usr/local/include /usr/include/hdf5/serial
   LIBRARY_DIRS:=$(PYTHON_LIB) /usr/local/lib /usr/lib/usr/lib/x86_64-linux-gnu/hdf5/serial
   

如果要用opencv3编译的话还需要改动:

    LIBRARIES += glog gflags protobuf boost_system boost_filesystem m hdf5_hl hdf5

    # handle IO dependencies
    USE_LEVELDB ?= 1
    USE_LMDB ?= 1
    USE_OPENCV ?= 1

    ifeq ($(USE_LEVELDB), 1)
        LIBRARIES += leveldb snappy
    endif
    ifeq ($(USE_LMDB), 1)
        LIBRARIES += lmdb
    endif
    ifeq ($(USE_OPENCV), 1)
        LIBRARIES += opencv_core opencv_highgui opencv_imgproc 

        ifeq ($(OPENCV_VERSION), 3)
            LIBRARIES += opencv_imgcodecs
        endif

    endif

4. 编译caffe

mkdir build

cd build

cmake ..

make all -j8

1. 编译pycaffe
   
   cd caffe-master
   
   make pycaffe
   

2. 编译测试模块
   
   make all
   
   make test
   
   make runtest
   

测试成功显示结果(部分):

[———-] Global test environment tear-down

[==========] 2091 tests from 283 test cases ran. (415487 ms total)

[ PASSED ] 2091 tests.

添加环境变量:

在~/.bashrc中添加

export PYTHONPATH=/home/hyzhan/caffe-master/:$PYTHONPATH

export PYTHONPATH=/home/hyzhan/caffe-master/caffe/python:$PYTHONPATH