【Caffe】ubuntu14.04下CPU版caffe配置小结

版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/signal926/article/details/52270668


前段时间在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/

猜你喜欢

转载自blog.csdn.net/signal926/article/details/52270668