Ubuntu 16.04配置caffe-ssd

1
2
3
4
5
6
sudo apt-get install libprotobuf-dev libleveldb-dev libsnappy-dev libopencv-dev
sudo apt-get install libhdf5-serial-dev protobuf-compiler
sudo apt-get install --no-install-recommends libboost-all-dev
sudo apt-get install libatlas-base-dev
sudo apt-get install python-dev
sudo apt-get install libgflags-dev libgoogle-glog-dev liblmdb-dev

1
2
3
git clone https://github.com/weiliu89/caffe.git
cd caffe
git checkout ssd

1
2
cd caffe
cp Makefile.config.example Makefile.config

1
2
3
4
5
6
7
8
9
1.若使用cudnn，取消“# USE_CUDNN := 1” 前的注释即：USE_CUDNN := 1
2.若使用opencv3.x，取消“# OPENCV_VERSION := 3” 前的注释，即：OPENCV_VERSION := 3
3.取消“# WITH_PYTHON_LAYER := 1” 前的注释。即 WITH_PYTHON_LAYER := 1
4.加入hdf5的目录：
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

1
2
3
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

1
2
3
NVCCFLAGS +=-ccbin=$(CXX) -Xcompiler-fPIC $(COMMON_FLAGS)
修改为：
NVCCFLAGS += -D_FORCE_INLINES -ccbin=$(CXX) -Xcompiler -fPIC $(COMMON_FLAGS)

1
2
3
LIBRARIES += boost_thread stdc++后加boost_regex
修改为:
LIBRARIES += boost_thread stdc++ boost_regex

1
2
3
4
make -j8
make py
make test -j8
make runtest -j8

1
2
3
4
5
6
7
8
9
10
CUDA_ARCH := -gencode arch=compute_20,code=sm_20 
-gencode arch=compute_20,code=sm_21 
-gencode arch=compute_30,code=sm_30 
-gencode arch=compute_35,code=sm_35 
-gencode arch=compute_50,code=sm_50 
-gencode arch=compute_52,code=sm_52 
-gencode arch=compute_61,code=sm_61
删除这两行即可：
-gencode arch=compute_20,code=sm_20 
-gencode arch=compute_20,code=sm_21 

1
sudo apt-get install libopenblas-dev

1
sudo apt-get install python-numpy

1
sudo apt-get install libtcmalloc-minimal4

1
export PYTHONPATH=~/caffe-ssd/python:$PYTHONPATH

1
export LD_PRELOAD="/usr/lib/libtcmalloc_minimal.so.4" 

1
pip install protobuf

1
pip install -U scikit-image

1
2
3
PYTHON_LIBRARIES := boost_python3 python3.5m
PYTHON_INCLUDE := /usr/include/python3.5m 
/usr/lib/python3/dist-packages/numpy/core/include

1
2
3
sudo ln -s
/usr/lib/x86_64-linux-gnu/libboost_python-py35.so
/usr/lib/x86_64-linux-gnu/libboost_python3.so

1
sudo apt-get install python3-skimage