Advertisement

ubuntu14.04 + caffe + python2.7 + CPU安装指南

阅读量:

本文基于ubuntu14.04 安装cpu版caffe,并安装pycaffe, 方便python调用caffe。
首先安装必须的依赖包:

复制代码
    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

我们要使用python调用caffe, 还需要python-dev包依赖:

复制代码
    sudo apt-get install python-dev

对于cpu, 我们需要安装加速cpu计算的数学依赖库:

复制代码
    sudo apt-get install libatlas-base-dev

对于ubuntu14.04 ,还需要安装如下依赖:

复制代码
    sudo apt-get install libgflags-dev libgoogle-glog-dev liblmdb-dev

caffe需要安装图像包依赖opencv开源库:

复制代码
    sudo apt-get install python-opencv

依赖包安装完成后,下面开始编译caffe。从caffe项目主页把caffe项目clone下来:

复制代码
    git clone --recursive https://github.com/BVLC/caffe.git

然后:

复制代码
    cd caffe
    cp Makefile.config.example Makefile.config

由于我们是无gpu安装,需要修改Makefile相关配置:

复制代码
    去掉CPU_ONLY :=1的注释
    注释掉CUDA有关的行:
    #CUDA_DIR := /usr/local/cuda
    #CUDA_DIR := /usr
    #CUDA_ARCH := ....
    #TEST_GPUID := 0
    去掉WITH_PYTHON_LAYER := 1的注释

修改完成后:

复制代码
    make all 
    make test
    make runtest

如果编译过程中没有出现错误,则编译成功。接下来编译pycaffe:

复制代码
    make pycaffe

完成后,

复制代码
    cd caffe/python
    打开python解释器, 若import caffe成功, 则安装成功。

否则:

复制代码
    make clean

重新编译。

参考:
【1】http://caffe.berkeleyvision.org/install_apt.html
【2】https://github.com/BVLC/caffe

全部评论 (0)

还没有任何评论哟~