ubuntu16.04-py35安装caffe

安装驱动、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/serial

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