Ubuntu16.04安装caffe(仅cpu)
阅读原文时间:2021年04月20日阅读:1

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

选择数学计算的库一般包括atlas,mkl,openblas。Mkl要收费,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相关依赖包

因为安装caffe对python的要求至少是2.7版本,Ubuntu 16.04系统自带了python2.7.12版本。分两步:下载安装pip和python-**依赖包;再用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 就行了。

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

$ 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.12(default,NOV 19 2016,06:48:10)

[GCC 5.4.0 20160619]on linux2

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

>>>import caffe

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

手机扫一扫

移动阅读更方便

阿里云服务器
腾讯云服务器
七牛云服务器

你可能感兴趣的文章