caffe框架安装

git clone https://github.com/BVLC/caffe.git

Compilation with Make

1. 复制Makefile.config

   sudo cp Makefile.config.example Makefile.config
   

2. 修改Makefile.config
   • 打开cudnn

     将
     #USE_CUDNN := 1
     修改成： 
     USE_CUDNN := 1
     

   • 指定opencv版本

     将
     #OPENCV_VERSION := 3 
     修改为： 
     OPENCV_VERSION := 3
     

   • 使用python接口

     将
     #WITH_PYTHON_LAYER := 1 
     修改为 
     WITH_PYTHON_LAYER := 1
     

   • 修改python路径

     INCLUDE_DIRS := $(PYTHON_INCLUDE) /usr/local/include
     LIBRARY_DIRS := $(PYTHON_LIB) /usr/local/lib /usr/lib 
     修改为： 
     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 /usr/lib/x86_64-linux-gnu/hdf5/serial
     

3. 修改Makefile
   • 改NVCCFLAGS

     将：
     NVCCFLAGS +=-ccbin=$(CXX) -Xcompiler-fPIC $(COMMON_FLAGS)
     替换为：
     NVCCFLAGS += -D_FORCE_INLINES -ccbin=$(CXX) -Xcompiler -fPIC $(COMMON_FLAGS)
     

   • 改LIBRARIES

     将：
     LIBRARIES += glog gflags protobuf boost_system boost_filesystem m hdf5_hl hdf5
     改为：
     LIBRARIES += glog gflags protobuf boost_system boost_filesystem m hdf5_serial_hl hdf5_serial
     

4. make

   make all
   make test
   make runtest
   

CMake Build

mkdir build
cd build
cmake ..
make all
make install
make runtest