下载Caffe
直接使用Git下载caffe,如果没有Git,请先安装Git:
1
| sudo apt-get install git
|
安装Caffe:
1
| git clone git://github.com/BVLC/caffe.git
|
安装依赖包
1 2 3 4 5 6 7 8 9 10 11
| sudo apt-get install libatlas-base-dev sudo apt-get install libprotobuf-dev sudo apt-get install libleveldb-dev sudo apt-get install libsnappy-dev sudo apt-get install libopencv-dev sudo apt-get install libboost-all-dev sudo apt-get install libhdf5-serial-dev sudo apt-get install libgflags-dev sudo apt-get install libgoogle-glog-dev sudo apt-get install liblmdb-dev sudo apt-get install protobuf-compiler
|
安装OpenCV(可选)
1 2 3 4 5 6
| cd caffe sudo git clone https://github.com/jayrambhia/Install-OpenCV cd Install-OpenCV/Ubuntu sudo sh dependencies.sh cd 2.4 sudo sh opencv2_4_10.sh
|
修改Caffe配置文件
1 2
| cd ~/caffe sudo cp Makefile.config.example Makefile.config
|
修改配置文件,本人使用vim,若果没有,请先安装:
1
| sudo apt-get install vim
|
打开文件:
修改如下:
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28
| //如果你不使用GPU的话,就将 # CPU_ONLY := 1 修改成: CPU_ONLY := 1
//若使用cudnn,则将 # USE_CUDNN := 1 修改成: USE_CUDNN := 1
//若使用的opencv版本是3的,则将 # OPENCV_VERSION := 3 修改为: OPENCV_VERSION := 3
//若要使用python来编写layer,则需要将 # WITH_PYTHON_LAYER := 1 修改为 WITH_PYTHON_LAYER := 1
//重要的一项!!! 将# Whatever else you find you need goes here.下面的 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 //这是因为ubuntu16.04的文件包含位置发生了变化,尤其是需要用到的hdf5的位置,所以需要更改这一路径
|
编译
1 2 3
| make all -j16("‐j16"是使用 CPU 的多核进行编译,可以极大地加速编译的速度) make test -j16 make runtest -j16
|