Ubuntu16.04安装caffe(仅cpu)

时间:2021-05-05 06:30:26

Ubuntu16.04安装caffe(仅cpu

背景:

之前用的centos6.6安装caffe,由于各种依赖的问题,并且源里面的安装包有限,有时需要去网站上下载包,涉及到对应的版本型号比较细致且复杂,最终centos安装caffe的路没走下去,选择了ubuntu16.04版本安装caffe

 

安装步骤如下(参考网络安装教程):

1 安装开发依赖包

 

$ sudo apt-get install build-essential

$ sudo apt-get install vim cmake git

$ sudo apt-get install libprotobuf-dev libleveldb-dev libsnappy-dev libopencv-dev libboost-all-dev libhdf5-serial-dev libgflags-dev libgoogle-glog-dev liblmdb-dev protobuf-compiler

 

2. 安装ATLAS

选择数学计算的库一般包括atlasmklopenblasMkl要收费,atlas安装简单。

$ sudo apt-get install libatlas-base-dev 

 

3. 下载caffe

把事先在gitub网站上下载好的caffe压缩包,

放在工作目录:home/yangxing/workspace/下,并在此解压。

$  unzip caffe-master.zip

 

或者直接在home/yangxing/workspace打开终端,用git命令下载

$ git clone https://github.com/BVLC/caffe-master.git  

 

如果报错,没有git命令,则需下载

$ apt-get install git

 

再执行上述git命令下载caffe

 

4. 安装Python相关依赖包

因为安装caffepython的要求至少是2.7版本,Ubuntu 16.04系统自带了python2.7.12版本。分两步:下载安装pippython-**依赖包;再用pip下载安装requirements.txt里的包。

 

首先需要在caffe-master目录下安装pip,

$ sudo apt-get install python-pip

 

由于pip 待会儿下载安装文件里的python依赖库时,需要pip 9.0.1版本

所以需要将pip升级到此版本。

$ pip install --upgrade pip

 

 

进入caffe-master目录下的python,安装scipy

 

$ cd/home/yangxing/workspace/caffe-master/python

$ sudo apt-get install python-numpy python-scipy python-matplotlib ipython ipython-notebook python-pandas python-sympy python-nose

 

最后同样在home/yangxing/workspace/caffe-master/python目录下,安装requirements.txt里面的包,需要切换到root权限

 

$ su

密码:****

$ pip install -r requirements.txt

$ for req in $ (cat requirements.txt); do pip install $ req; done

如果提示报错,一般是缺少必须的包引起的,直接根据提示 

$ pip install <package-name>就行了。

如果你不知道具体哪个没有安装,你可以执行下面命令进行检查

$ pip install -r requirements.txt

安装完后退出root权限

 

$ su yangxing

 

 

4编译caffe

改配置安装好上述的依赖包后,就开始编译安装caffe

首先修改配置文件,回到caffe目录下

 

$ cd ~/caffe-master

$ cp Makefile.config.example  Makefile.config

$ udo gedit  Makefile.config

 

 

修改如下:

cpu_only =1取消注释,(即去掉#符号)

USE_LMDB :=1取消注释

BLAS =atlas取消注释

修改python包目录,这句话

PYTHON_INCLUDE := /usr/include/python2.7 \

  /usr/lib/python2.7/dist-packages/numpy/core/include

改为

PYTHON_INCLUDE := /usr/include/python2.7 \

  /usr/local/lib/python2.7/dist-packages/numpy/core/include

 

为了使caffe编译时候能正确调用python,需要在系统内添加caffe里的python地址,我们执行下面的命令

 

$ sudo gedit ~/.bashrc

在文件的最下面添加caffe的地址

 

 #caffe

export PYTHONPATH=/home/yangxing/workspace/caffe-master/python

 

添加成功后直接按保存后退出

让刚刚修改的路径立即生效,执行命令:

 

$ source ~/.bashrc

 

开始编译caffe了,在/home/yangxing/workspace/caffe-master下,用make

 

$ make all -j16   #视自己的电脑来选择-j8或者-j4,我电脑可以-j16,编译速度快些

$ make test

$ make runtest

$ make pycaffe

 

检验python的接口环境是否配置成功,在caffe下,输入python,在python的编辑下import caffe,如果不出错,表明python接口配置完成。

 

$ cd /home/yangxing/workspace/caffe-master

$ python

Python 2.7.12defaultNOV 19 201606:48:10

[GCC 5.4.0 20160619]on linux2

Type “help”,”copyright”,”credits”or “licence” for more information.  

>>>import caffe

 

如果不报错,那就说明安装好了。