安装依赖项
sudo apt-get install libprotobuf-dev libleveldb-dev libsnappy-dev libopencv-dev libhdf5-serial-dev protobuf-compiler
sudo apt-get install --no-install-recommends libboost-all-dev
sudo apt-get install libatlas-base-dev
sudo apt-get install libhdf5-serial-dev
sudo apt-get install libgflags-dev libgoogle-glog-dev liblmdb-dev
sudo apt-get install libblas-dev
安装opencv,详细方法可见本人文章opencv linux安装与编译.
python安装
caffe支持python2.7版本,其他版本是否支持不太清楚,网络上给的安装方法基本都是基于python2.7版本上的.
python --version
查看系统默认python版本.
安装python-dev依赖包
sudo apt-get install python-dev
安装caffe
下载caffe
sudo apt-get install git
git clone git://github.com/BVLC/caffe.git
编译caffe
cd caffe/
cp Makefile.config.example Makefile.config
sudo vim Makefile.config
对Makefile.config文件做如下修改:
因为CPU MODE, 所以在CPU_ONLY := 1前面的#要去掉.
在opencv 3.x版本下编译,所以在OPENCV_VERSION := 3前面的#要去掉.
-
两个路径要改成这样:(添加后面的两个hdf5的路径, 否则编译时报hdf5错误)
# 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/hdf5/serial
下面开始编译:
make all
make test
make runtest
这个make默认是用CPU单核运算,如果想要快一点, 比如我想使用四核, 在make后面加上-j4标签.
如果上面4行某一行报错之后想要重试,建议先make clean再重新开始.
编译python接口
通常为了更好的调用caffe,我们往往需要写代码,caffe具有Python和C++接口,我们较为常用Python接口,这里我将展示如何编译Python接口。
安装pip
sudo apt-get install python-pip
安装Python接口依赖库.在caffe根目录下,有个python文件夹,文件夹里面有个requirements.txt,里面有需要的依赖库和版本信息,按照其安装即可,在安装前,需要先安装fortran编辑器(gfrotran),因为安装scipy库时需要它,命令如下:
sudo apt-get install gfortran
cd ~/caffe/python
for req in $(cat requirements.txt); do pip install $req; done
安装结束后,可以执行如下语句验证:
sudo pip install -r requirements.txt
可以看到,安装成功的,都会显示Requirement already satisfied,没安装成功的,会继续进行安装。
将caffe根目录下的python文件夹加入到环境变量
先打开配置文件bashrc
sudo vim ~/.bashrc
在文件的最后面添加
export PYTHONPATH=/home/robert/Documents/caffe/python:$PYTHONPATH
然后执行更新配置(有时更新不起作用时,可以重启电脑就会有效):
sudo ldconfig
编译python接口
cd ../../caffe/
make pycaffe
验证python接口
python
import caffe
如果import成功,则安装成功.