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 <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.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
如果不报错,那就说明安装好了。