caffe安装过程

目前使用的caffe是ssd的作者weiliu自己改过的caffe，可能与原版caffe不太一样。

 

一、安装各种依赖项

sudo apt-get update sudo apt-get install -y build-essential cmake git pkg-config sudo apt-get install -y libprotobuf-dev libleveldb-dev libsnappy-dev libhdf5-serial-dev protobuf-compiler sudo apt-get install -y libatlas-base-dev libopenblas-dev sudo apt-get install -y --no-install-recommends libboost-all-dev sudo apt-get install -y libgflags-dev libgoogle-glog-dev liblmdb-dev sudo apt-get install -y python-pip sudo apt-get install -y python-dev sudo apt-get install -y python-numpy python-scipy

sudo apt-get install -y liblapack-dev libblas-dev  

 

二、安装requirement.txt

cd /home/hd/caffe/python

打开requirements.txt 删掉iPython那一句，因为iPython 6.0已经不支持python2.7了，所以不要从这里来安装了

继续输入：pip install -r requirements.txt

等待安装完成

 

三、编译caffe

cd ~/caffe

sudo cp Makefile.config.example Makefile.config

打开Makefile.config：sudo gedit Makefile.config

 

如果是用CPU，删掉# CPU_ONLY := 1的注释

如果是用GPU，删掉# USE_CUDNN := 1的注释

 

因为用的是opencv3，所以删掉# OPENCV_VERSION := 3的注释

 

把Whatever else you find you need goes here.下面改成

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

（因为ubuntu 16.04把包的目录改了一下，所以需要做一些更新）

 

若要使用python来编写layer，则需要将# WITH_PYTHON_LAYER := 1的注释删掉

 

打开Makefile：sudo gedit Makefile

将 NVCCFLAGS +=-ccbin=$(CXX) -Xcompiler-fPIC $(COMMON_FLAGS)

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

 

之后再

make all -j8

make pycaffe

 

编译结束后，添加环境变量

sudo gedit ~/.bashrc

在末尾添加export PYTHONPATH=/home/wanghh/caffe/python:$PYTHONPATH

source ~/.bashrc

大功告成。