您的位置:首页 > Web前端

虚拟机下Ubuntu系统安装Caffe(CPU版本)详细过程

2018-02-27 18:47 609 查看
博主按照下面的参考博客一步步安装,亲测有效。一定不要着急,安装过程也可能出现其它问题,通过百度、谷歌这些问题总会找到解决办法。综合楼主遇到的问题,重新梳理一下参考博客的内容。

参考博客:深度学习文章1:Caffe安装教程:Ubuntu16.04(CPU)

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

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

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

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

ctrl+alt+T打开终端,输入:

lspci | grep -i nvidia


结果1:如果未显示任何内容,则证明你的电脑不支持GPU。

结果2:如果显示如下图内容:



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

二、依次安装依赖包

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

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的路径问题)

原来的路径配置如下图:



修改为如下图:



新增的内容为:

/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接口依赖库

由于按照参考博客,出现了下面提示的问题:* (gedit:6191): WARNING *: Set document metadata failed: 不支持设置属性 metadata::gedit-spell-enabled

所以,这里相比参考博客,增加一个步骤。按ctrl+alt+T打开另一个终端,输入:

sudo apt-get install lib32stdc++6


如下图,其中:在您希望继续执行吗? [Y/n] Y。



这个问题就解决了。

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

cd caffe/
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/
sudo make pycaffe


5)验证python接口

进行python环境,引入caffe包,如果没有报错则安装成功!

如下图:



至此,caffe安装成功!

六、题外话

因为系统或者安装机器的不同,可能会出现一些不一样的问题,出现问题,请粘贴问题自行google或百度,都是有答案的,切忌眼高手低。
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息