红联Linux门户
Linux帮助

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

发布时间:2016-11-22 10:44:08来源:linux网站作者:莫欺丶少年穷
本文是在新安装的Ubuntu16.04上进行Caffe的安装,每个版本的Ubuntu安装略有不同,请自行搜索。
 
一、检查自己电脑是否具有GPU
通常Caffe在计算时有两种模式可以选择,CPU或GPU,使用GPU处理图像速度会更快,但往往有的计算机没有GPU,配置太低,所以只能选择CPU,作者的电脑不支持GPU,因此选择CPU安装的版本。
输入:
lspci | grep -i nvidia
结果1
如果未显示任何内容,则证明你的电脑不支持GPU。
结果2
如果显示如下图内容:
Caffe安装教程:Ubuntu16.04(CPU)
则表明版本是GTX 650,支持GPU,可以去http://developer.nvidia.com/cuda-gpus去验证,支持CUDA,本文只描述CPU模式的caffe安装,如果要安装GPU模式的caffe,请自行搜索。
 
二、依次安装依赖包
依次执行以下语句,安装依赖包:
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安装教程:Ubuntu16.04(CPU)
 
四、编译caffe
1.进入caffe目录
cd caffe/
2.生成Makefile.config文件,这里是将caffe目录下自带的Makefile.config.example文件复制一份并更名为Makefile.config,命令如下:
cp Makefile.config.example Makefile.config
此时目录下就会有Makefile.config文件,如下图:
Caffe安装教程:Ubuntu16.04(CPU)
3.修改Makefile.config文件中的配置
1)编辑Makefile.config文件
sudo gedit Makefile.config
2)去掉CPU_ONLY前面的#号
Caffe安装教程:Ubuntu16.04(CPU)
使得CPU_ONLY := 1
3)配置引用文件路径(主要是HDF5的路径问题)
原来的路径配置如下图:
Caffe安装教程:Ubuntu16.04(CPU)
修改为如下图:
Caffe安装教程:Ubuntu16.04(CPU)
新增的内容为:
/usr/include/hdf5/serial
/usr/lib/x86_64-linux-gnu/hdf5/serial
4)执行编译
sudo make all
sudo make test
sudo make runtest
执行玩runtest后,会有结果显示:
Caffe安装教程:Ubuntu16.04(CPU)
显示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安装教程:Ubuntu16.04(CPU)
注意,这里路径根据每个人电脑则不一样,例如,我的下载的caffe在/home/moqi/下面,所以路径为/home/moqi/caffe/python,你可以自行进入自己的caffe文件夹里的python目录下,输入pwd查看路径。
然后执行更新配置(有时更新不起作用时,可以重启电脑就会有效):
sudo ldconfig
4)编译python接口
cd ~/caffe/
make pycaffe
5)验证python接口
进行python环境,引入caffe包,如果没有报错则安装成功!
如下图:
Caffe安装教程:Ubuntu16.04(CPU)
至此,caffe安装成功!
 
六、题外话
因为系统或者安装机器的不同,可能会出现一些不一样的问题,出现问题,请粘贴问题自行搜索,都是有答案的,切忌眼高手低。
 
本文永久更新地址:http://www.linuxdiyf.com/linux/26229.html