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

参考:https://blog.csdn.net/u010193446/article/details/53259294



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

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

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

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

输入:

lspci | grep -i nvidia
  
  
  • 1
结果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
  
  
  • 1
  • 2
  • 3
  • 4
  • 5
  • 6
  • 7
  • 8
  • 9
  • 10

三、下载caffe

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

sudo apt-get install git
  
  
  • 1

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

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

这样你就可以在根目录下看到有个caffe目录了
这里写图片描述

四、编译caffe

1.进入caffe目录

cd caffe/
  
  
  • 1

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

cp Makefile.config.example Makefile.config
  
  
  • 1

此时目录下就会有Makefile.config文件,如下图:
这里写图片描述

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

1)编辑Makefile.config文件

sudo gedit Makefile.config
  
  
  • 1

2)去掉CPU_ONLY前面的#号

这里写图片描述

使得CPU_ONLY := 1

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

原来的路径配置如下图:

这里写图片描述

修改为如下图:

这里写图片描述

新增的内容为:

注意添加前面的空格,否则编译会提示找不到 hdf5.h

/usr/include/hdf5/serial
/usr/lib/x86_64-linux-gnu/hdf5/serial
  
  
  • 1
  • 2

4)执行编译

sudo make all
sudo make test
sudo make runtest
  
  
  • 1
  • 2
  • 3

执行玩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
  
  
  • 1

2)安装Python接口依赖库

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

sudo apt-get install gfortran
cd ~/caffe/python
for req in $(cat requirements.txt); <span class="hljs-keyword">do</span> pip install <span class="hljs-variable">$req; done
  
  
  • 1
  • 2
  • 3

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

sudo pip install -r requirements.txt
  
  
  • 1

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

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

先打开配置文件bashrc

sudo gedit ~/.bashrc
  
  
  • 1

在文件的最后面添加

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

如下图:
这里写图片描述

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

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

sudo ldconfig
  
  
  • 1

4)编译python接口

cd ~/caffe/
make pycaffe
  
  
  • 1
  • 2

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

至此,caffe安装成功!



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

猜你喜欢

转载自blog.csdn.net/unfound/article/details/82020265
今日推荐