ubuntu14.04下CPU版caffe配置小结

来源:互联网 发布:软件退税流程图 编辑:程序博客网 时间:2024/05/18 00:43

  • 我的配置过程无GPU版
    • 安装依赖项
      • 依赖项说明
    • 下载caffe
    • 安装python依赖库
    • 配置OpenCV环境
    • 编译caffe
    • 添加环境变量
  • 补充帖
    • win7下配置caffe
    • 官方提供的caffe配置方法
    • 最简洁概要的caffeDIGITS配置


前段时间在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/pythonfor 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-OpenCVcd Install-OpenCV/RedHatsudo ./opencv_latest.sh

5.编译caffe:

根据需要修改配置文件:

cp Makefile.config.example Makefile.configsudo gedit Makefile.config

此处没有使用GPU和cuDNN,所以取消CPU_ONLY := 1前面的注释即可。
然后开始编译caffe:

make allmake testmake 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:$PYTHONPATHexport 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/

0 0
原创粉丝点击