前段时间在win7、ubuntu14.04下都配置过caffe,结合配置流程,把踩过的坑都记录一下,以供大家参考。以下主要为ubuntu14.04下CPU版的配置过程,其他配置过程可参考最后的补充帖(均已验证过)。
我的配置过程:(无GPU版)
1.安装依赖项:
sudo apt-get install libprotobuf-dev libleveldb-dev libsnappy-dev libopencv-dev libboost-all-dev libhdf5-serial-dev libgflags-dev libgoogle-glog-dev liblmdb-dev protobuf-compiler libatlas-base-dev
依赖项说明:
主要用到的dependency:
glog:打印调试信息,这个对于调错很有用。
gflags:命令行参数
gtest:测试框架
protobuf:数据的序列化
boost:一些类似C++11的feature,因为早期cuda不支持c++11
opencv:图像处理函数
leveldb,lmdb:简单的本地数据库。
cuda:GPU计算框架
atlas/mkl/eigen:线性代数计算库
2.下载caffe:
sudo git clone git://github.com/BVLC/caffe.git
3.安装python依赖库:
查看“requirements.txt”文件,用pip工具依次安装python依赖库。(假设放置caffe的路径为$ROOT,根据个人情况修改路径)
注意:如果使用spyder IDE,需要先在干净的系统下安装spyder,再安装“requirements.txt”中的依赖项,否则可能因为numpy、scipy等版本冲突导致IDE无法打开。
cd /$ROOT/caffe/python
for req in $(cat requirements.txt); do sudo pip install $req; done
4.配置OpenCV环境
Caffe作者默认你已经配置好了OpenCV环境,文档里没有说这一步。好在有人已经写好了配置OpenCV的脚本, https://github.com/jayrambhia/Install-OpenCV ,直接拿来用。
git clone https://github.com/jayrambhia/Install-OpenCV
cd Install-OpenCV/RedHat
sudo ./opencv_latest.sh
5.编译caffe:
根据需要修改配置文件:
cp Makefile.config.example Makefile.config
sudo gedit Makefile.config
此处没有使用GPU和cuDNN,所以取消CPU_ONLY := 1
前面的注释即可。
然后开始编译caffe:
make all
make test
make runtest
可用多线程加速:
make all -j4
make test -j4
make runtest -j4
编译python接口:
sudo make pycaffe -j4
编译MATLAB接口:
sudo make matcaffe -j4
6. 添加环境变量:
sudo gedit ~/.bashrc
在文本最后添加这两句:
export PYTHONPATH=/$ROOT/caffe/python:$PYTHONPATH
export CAFFE_HOME=/$ROOT/caffe
更新环境变量:
source .bashrc
(注意不要漏掉bashrc前面的点)
补充帖
win7下配置caffe:
https://github.com/Microsoft/caffe
官方提供的caffe配置方法:
http://caffe.berkeleyvision.org/installation.html
最简洁概要的caffe+DIGITS配置
(含CUDA+cuDNN+DIGITS)
http://blog.csdn.net/enjoyyl/article/details/47397505/