git clone https://github.com/BVLC/caffe.git
三、编译caffe
1)进入caffe目录:cd caffe/
2)生成Makefile.config文件:cp Makefile.config.example Makefile.config
3)修改Makefile.config文件中的配置
a) 编辑Makefile.config文件:sudo gedit Makefile.config
b) 去掉CPU_ONLY前面的#号
c) 配置引用文件路径(主要是HDF5的路径问题:hdf5.h、 libhdf5.so的路径)INCLUDE_DIRS := $(PYTHON_INCLUDE) /usr/local/include/usr/include/hdf5/serial
LIBRARY_DIRS := $(PYTHON_LIB) /usr/local/lib /usr/lib/usr/lib/i386-linux-gnu/hdf5/serial
注意:可以利用sudo find / -name 文件名来查找某个文件的具体位置
d) 执行编译执行完runtest后,如果显示PASSED,则执行成功,如下图:
注意:如果执行结果失败,则需要执行语句sudo make clean后,解决掉问题,重新编译。
四、编译python接口
1) 安装pip
Ubuntu16.04默认安装的是python2.7,没有pip,所以需要先安装pip,命令如下:sudo apt-get install python-pip
2) 安装Python接口依赖库
在caffe根目录下,有个python文件夹,该文件夹有requirements.txt,里面有需要的依赖库和版本信息,按照其安装即可:
cd ~/caffe/python
for req in $(cat requirements.txt); do pip install $req; done
安装结束后,可以执行如下语句验证:sudo pip install -r requirements.txt,如果显示Requirement already satisfied,则安装成
功,否则会继续进行安装。
3) 将caffe根目录下的python文件夹加入到环境变量中
a) 打开配置文件bashrc:sudo gedit ~/.bashrc
b) 在文件的最后面添加:export PYTHONPATH=/home/shine/caffe/python:$PYTHONPATH
注意:进入caffe文件夹里的python目录下,输入pwd就可以查看路径
c) 执行更新配置(有时更新不起作用时,重启电脑就会有效):sudo ldconfig
4)编译python接口
cd ~/caffe/
sudo make pycaffe
注意:如果出现找不到numpy/arrayobject.h这种问题,则检查Makefile.config文件中的PYTHON路径(Python.h、
numpy/arrayobject.h的路径):
PYTHON_INCLUDE := /usr/include/python2.7 /usr/lib/python2.7/dist-packages/numpy/core/include
5) 验证python接口
进行python环境,引入caffe包,如果没有报错,则安装成功!如下图: