美文网首页
Caffe安装教程:Ubuntu16.04(CPU only)

Caffe安装教程:Ubuntu16.04(CPU only)

作者: 生活科技 | 来源:发表于2017-12-11 11:32 被阅读0次

    Caffe安装教程:Ubuntu16.04(CPU)

    本文是在新安装的Ubuntu16.04上进行Caffe的安装,每个版本的Ubuntu安装略有不同,请自行百度。

    一、检查自己电脑是否具有GPU

    通常Caffe在计算时有两种模式可以选择,CPU或GPU,使用GPU处理图像速度会更快,但往往有的计算机没有GPU,配置太低,所以只能选择CPU,作者的电脑不支持GPU,因此选择CPU安装的版本。

    输入:

    lspci | grep -i nvidia
    

    <dl>

    <dt>结果1</dt>

    <dd>如果未显示任何内容,则证明你的电脑不支持GPU。</dd>

    <dt>结果2</dt>

    <dd>如果显示如下图内容:


    这里写图片描述

    则表明版本是GTX 650,支持GPU,可以去http://developer.nvidia.com/cuda-gpus去验证,支持CUDA,本文只描述CPU模式的caffe安装,如果要安装GPU模式的caffe,请自行百度或Google。</dd>

    </dl>

    二、依次安装依赖包

    依次执行以下语句,安装依赖包:

    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 libgflags-dev
    sudo apt-get install libgoogle-glog-dev
    sudo apt-get install liblmdb-dev
    sudo apt-get install libatlas-base-dev
    

    三、下载caffe

    可直接使用Git下载caffe,如果没有Git,请先安装Git,作者新装的Ubuntu16.04没有Git,所以先安装Git,执行如下语句:

    sudo apt-get install git
    

    安装结束后下载caffe,执行以下命令:

    git clone git://github.com/BVLC/caffe.git
    

    这样你就可以在根目录下看到有个caffe目录了


    这里写图片描述

    四、编译caffe

    1.进入caffe目录

    cd caffe/
    

    2.生成Makefile.config文件,这里是将caffe目录下自带的Makefile.config.example文件复制一份并更名为Makefile.config,命令如下:

    cp Makefile.config.example Makefile.config
    

    此时目录下就会有Makefile.config文件,如下图:


    这里写图片描述

    3.修改Makefile.config文件中的配置

    1)编辑Makefile.config文件

    sudo gedit Makefile.config
    

    2)去掉CPU_ONLY前面的#号

    这里写图片描述

    使得CPU_ONLY := 1

    3)配置引用文件路径(主要是HDF5的路径问题)

    原来的路径配置如下图:

    [图片上传失败...(image-b60a7f-1512963090846)]

    修改为如下图:

    这里写图片描述

    新增的内容为:

    /usr/include/hdf5/serial
    /usr/lib/x86_64-linux-gnu/hdf5/serial
    

    4)执行编译

    sudo make all
    sudo make test
    sudo make runtest
    

    执行玩runtest后,会有结果显示:

    这里写图片描述

    显示1098个用例执行成功,结果PASSED!

    注意:如果执行结果失败,则需要执行语句sudo make clean,然后解决掉问题重新编译

    五、编译python接口

    通常为了更好的调用caffe,我们往往需要写代码,caffe具有Python和C++接口,我们较为常用Python接口,这里我将展示如何编译Python接口。

    1)安装pip
    因为Ubuntu16.04默认安装的是python2.7,没有pip,所以需要先安装pip,命令如下:

    sudo apt-get install python-pip
    

    2)安装Python接口依赖库

    在caffe根目录下,有个python文件夹,文件夹里面有个requirements.txt,里面有需要的依赖库和版本信息,按照其安装即可,在安装前,需要先安装fortran编辑器(gfrotran),因为安装scipy库时需要它,命令如下:

    sudo apt-get install gfortran
    cd ~/caffe/python
    for req in $(cat requirements.txt); do pip install $req; done
    

    安装结束后,可以执行如下语句验证:

    sudo pip install -r requirements.txt
    

    可以看到,安装成功的,都会显示Requirement already satisfied,没安装成功的,会继续进行安装。

    3)将caffe根目录下的python文件夹加入到环境变量

    先打开配置文件bashrc

    sudo gedit ~/.bashrc
    

    在文件的最后面添加

    export PYTHONPATH=/home/moqi/caffe/python:$PYTHONPATH
    

    如下图:


    这里写图片描述

    注意,这里路径根据每个人电脑则不一样,例如,我的下载的caffe在/home/moqi/下面,所以路径为/home/moqi/caffe/python,你可以自行进入自己的caffe文件夹里的python目录下,输入pwd查看路径。

    然后执行更新配置(有时更新不起作用时,可以重启电脑就会有效):

    sudo ldconfig
    

    4)编译python接口

    cd ~/caffe/
    make pycaffe
    

    5)验证python接口
    进行python环境,引入caffe包,如果没有报错则安装成功!
    如下图:


    这里写图片描述

    至此,caffe安装成功!

    六、题外话

    因为系统或者安装机器的不同,可能会出现一些不一样的问题,出现问题,请粘贴问题自行google或百度,都是有答案的,切忌眼高手低。

    相关文章

      网友评论

          本文标题:Caffe安装教程:Ubuntu16.04(CPU only)

          本文链接:https://www.haomeiwen.com/subject/dohiixtx.html