安装驱动、cuda9.0、cudnn7
1.安装依赖
sudo apt-get install -y libprotobuf-dev libleveldb-dev libsnappy-dev libopencv-dev libhdf5-serial-dev protobuf-compiler libgflags-dev libgoogle-glog-dev liblmdb-dev
sudo apt-get install -y --no-install-recommends libboost-all-dev
sudo apt-get install -y libatlas-base-dev python3-dev python3-pip git
sudo ln -s /usr/lib/x86_64-linux-gnu/libboost_python-py35.so /usr/lib/x86_64-linux-gnu/libboost_python3.so
2.克隆caffe
git clone https://github.com/BVLC/caffe.git
cd caffe/
3.复制编译模板文件
cp Makefile.config.example Makefile.config
修改Makefile.config,开启cudnn:
取消# USE_CUDNN := 1行前面的#改为
USE_CUDNN := 1
4.对于cuda大于等于9.0需要去掉
-gencode arch=compute_20,code=sm_20 \
-gencode arch=compute_20,code=sm_21 \
5.修改python版本,
PYTHON_INCLUDE := /usr/include/python2.7 \
/usr/lib/python2.7/dist-packages/numpy/core/include
将上面两行改为
PYTHON_LIBRARIES := boost_python3 python3.5m
PYTHON_INCLUDE := /usr/include/python3.5 \
/usr/lib/python3.5/dist-packages/numpy/core/include
6.将Whatever else you find you need goes here.下面的
INCLUDE_DIRS
:= $(PYTHON_INCLUDE) /usr/
local/include
LIBS
:= $(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/seri
al
7.修改Makefile 文件(注意这里是makefile上面改的是makefile.config)
打开makefile文件,做如下修改:
将:NVCCFLAGS +=-ccbin=$(CXX) -Xcompiler-fPIC $(COMMON_FLAGS)
替换为:
NVCCFLAGS += -D_FORCE_INLINES -ccbin=$(CXX) -Xcompiler -fPIC $(COMMON_FLAGS)
8.编译caffe
make all -j4
make test -j4
make runtest -j4
make pycaffe