Ubuntu16.04安装Caffe(CPU_ONLY)

     

      本文记录的是自己在Ubuntu16.04上配置caffe环境跟编译的全部过程。


1、安装依赖库

    sudo apt-get install libprotobuf-dev 
    sudo apt-get install libleveldb-dev 
    sudo apt-get install libsnappy-dev 
    sudo apt-get install libopencv-dev 
    sudo apt-get install libhdf5-serial-dev 
    sudo apt-get install protobuf-compiler
    sudo apt-get install --no-install-recommends libboost-all-dev
    接下来是BLAS:
     sudo apt-get install libatlas-base-dev
    使用默认Python来建立pycaffe接口,需要安装:
    sudo apt-get install python-dev
    一些兼容性依赖库:
    sudo apt-get install libgflags-dev
    sudo apt-get install libgoogle-glog-dev 
    sudo apt-get install liblmdb-dev

    sudo apt-get install python-skimage

    sudo apt-get install python-protobuf


2、下载Caffe源码
   

    使用git下载Caffe源码
    git clone https://github.com/BVLC/caffe.git

    如果git觉得下载太慢,可以直接复制此链接到浏览器,下载zip格式,解压后即是caffe的源码。

Ubuntu16.04安装Caffe(CPU_ONLY)_第1张图片
    如果需要Caffe的Python接口,切换到caffe下的python目录下,输入以下命令下载python依赖库(先安装pip):
    sudo apt-get install python-pip
    for req in $(cat requirements.txt); do pip install $req; done


3、编译Caffe


    到Caffe文件夹中,拷贝一份Makefile.config.example并重命名成Makefile.config,修改该配置文件:
    cp Makefile.config.example Makefile.config
    打开Makefile.config,所以去掉CPU_ONLY := 1前面的注释;
    由于Ubuntu16.04文件结构的变化,#Whatever else you find you need goes here.处要改成下面这样:


    # 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默认单核运算,如果想加快速度,我这里是4核,可以在每条命令后面加上-j4,如果有报错,建议最好make clean重新        开    始。
    如果所有测试都通过,则说明安装好了。

 

4、测试


    #1. 把环境变量路径放到 ~/.bashrc文件中
    sudo echo export PYTHONPATH="~/caffe/python" >> ~/.bashrc
    #2. 使环境变量生效
    source ~/.bashrc

    cd  caffe/python目录下,然后进入python环境,输入:
    import caffe
    如果没有报错,证明安装成功。
Ubuntu16.04安装Caffe(CPU_ONLY)_第2张图片

 

参考了以下链接:

官方安装文档:http://caffe.berkeleyvision.org/installation.html

官方Ubuntu安装文档:http://caffe.berkeleyvision.org/install_apt.html

博客:http://www.linuxidc.com/Linux/2016-09/135034.htm

           http://www.linuxdiyf.com/linux/23093.html
 

你可能感兴趣的:(ML,caffe)