1. 安装caffe依赖包
$ sudo apt-get install git
$ 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 libhdf5-serial-dev
$ sudo apt-get install protobuf-compiler
$ sudo apt-get install --no-install-recommends
$ sudo apt-get install libboost-all-dev
$ sudo apt-get install libatlas-base-dev
$ sudo apt-get install python-dev
$ sudo apt-get install libgflags-dev
$ sudo apt-get install libgoogle-glog-dev
$ sudo apt-get install liblmdb-dev
2. 下载caffe
$ git clone https://github.com/BVLC/caffe.git
3. 修改 配置文件
首先,在caffe根目录下复制一份配置文件:
$ cd caffe
$ cp Makefile.config.example Makefile.config
然后,修改文件内容:
①CPU;②PYTHON_LIBRARIES; ③PYTHON_INCLUDE; ④INCLUDE_DIRS;⑤LIBRARY_DIRS.
...
...
# CPU-only switch (uncomment to build without GPU support).
CPU_ONLY := 1
...
...
# Uncomment to use Python 3 (default is Python 2)
PYTHON_LIBRARIES := boost_python-py35 python3.5m
PYTHON_INCLUDE := /usr/include/python3.5m \
/usr/lib/python3/dist-packages/numpy/core/include
...
...
# 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
...
...
4. 添加python环境 变量
在主目录下输入命令:
$ vim ~/.bashrc
按 ‘a’ 进入编辑模式,在文档末尾添加下列语句:
# added by Rain Jeyin in 2018/7/26
export PYTHONPATH=/home/user/caffe/python:$PYTHONPATH
注意:user为自己的主机用户名
然后按 ‘ESC’ 退出编辑模式,输入 ':wq' 保存退出
在主目录下输入如下命令,使上述操作生效。
$ source ~/.bashrc
5. 开始编译
在caffe目录下完成:
$ make all -j4
$ make test -j4
$ make runtest -j4
$ make pycaffe
6. 验证
进入python, 导入caffe 模块
$ python
>>import caffe
7. 问题解决
报错了:ImportError: No module named 'google'
安装对应模块: