Caffe初学者第一部:Ubuntu14.04上安装caffe(CPU)+Python的详细过程 (亲测成功, 20180524更新)

时间:2021-05-11 14:31:35

前言:

最近在学习深度学习,最先要解决的当然是开源框架的环境安装了。之前一直在学习谷歌的Tensorflow开源框架,最近实验中需要跟别人的算法比较,下载的别人的代码很多都是Caffe的,所以想着搭建好Caffe环境跑别人的代码。这中间经历过很多弯路,开始是入了Ubuntu16.04的坑,很多教程都说GCC版本不匹配,需要降级,我也尝试过发现很多坑;另外,就是安装matlab版本的Caffe以及安装GPU版本的Caffe,都经历了很多的波折,这前后摸索大概花了半个月左右。最后发现Ubuntu14.04最适合安装Caffe环境。

于是,我想着既然GPU版本或者matlab版本的Caffe环境配置很多坑,为什么不先从最简单的CPU+Python版本开始配置呢?到后面对Caffe框架熟悉了或者有需要用matcaffe的时候,再去摸索GPU或者matlab版本的Caffe配置。这就是我写本博客的初衷,先写个配置Caffe CPU ONLY版本的博客教程,希望初学者不再对caffe环境配置谈虎色变。话不多说,开始我们的教程吧!

首先,说一下本教程适用的计算机系统环境,本人笔记本安装了Win7+Ubuntu14.04双系统。至于安装双系统,可以参考网上博客:https://www.jianshu.com/p/dfd9436956b9

我下载的Ubuntu14.04是Kyliin版本,这里附上下载链接,直接复制到迅雷就可以下载速度很快:http://cdimage.ubuntu.com/ubuntukylin/releases/14.04/release/ubuntukylin-14.04.5-desktop-amd64.iso

至于硬件嘛,由于安装的是CPU ONLY版本的Caffe,我觉得现在的计算机CPU基本上都可行吧,内存啥的也不是要求太高。

亲测教程:

至此,同学你应该装好了Ubuntu14.04了,对吧,那我继续手把手教你。

注意,此时装好Ubuntu14.04之后,不要换源,系统可能会提示你升级到Ubuntu16.04,你选择不升级。

此时,打开终端

Step1:输入以下代码:

sudo apt-get update

Step2:安装caffe必要依赖,记住每一步的依赖都要安装:

sudo apt-get install git

sudo apt-get install libprotobuf-dev libleveldb-dev libsnappy-dev libopencv-dev libhdf5-serial-dev protobuf-compiler

sudo apt-get install build-essential

sudo apt-get install --no-install-recommends libboost-all-dev

sudo apt-get install libatlas-base-dev

sudo apt-get install --no-install-recommends libboost-all-dev

sudo apt-get install libgflags-dev libgoogle-glog-dev liblmdb-dev

Step3:官网拉取源码:

cd
git clone https://github.com/BVLC/caffe.git

注意第三步如果终端下载太慢,可以将该网址

https://github.com/BVLC/caffe.git

复制到浏览器地址栏,然后手动点击下载ZIP包,“caffe-master.zip”,之后解压到home文件夹即可,将解压出来的文件夹重命名为“caffe”。

Step4:安装第二部分依赖:

sudo apt-get install libblas-dev liblapack-dev libatlas-base-dev gfortran python-numpy

sudo apt-get install python-pip

sudo apt-get install python-numpy python-scipy python-matplotlib python-sklearn 

sudo apt-get install python-skimage python-h5py python-protobuf python-leveldb 

sudo apt-get install python-networkx python-nose python-pandas 

sudo apt-get install python-gflags Cython ipython 

sudo pip install --upgrade matplotlib

sudo apt-get update

Step5:修改配置文件:

cd caffe

cp Makefile.config.example Makefile.config
gedit Makefile.config

仅修改一处,将“CPU_ONLY”选项打开即可。

(注意:如果是Ubuntu16.04,除了修改上处外,还需修改以下地方:

加入后面的部分即可

INCLUDE_DIRS := $(PYTHON_INCLUDE) /usr/local/include /usr/include/hdf5/serial
LIBRARY_DIRS := $(PYTHON_LIB) /usr/local/lib /usr/lib /usr/lib/x86_64-linux-gnu/hdf5/serial

)

Step6:编译及测试:

make all -j4

make test -j4
make runtest -j4
make pycaffe -j4
make distribute

测试时,输入以下代码:

cd /home/dylan/caffe/python
python
import caffe

其中”dylan“是我的Ubutnu用户名,更改为你自己的即可,如果不出错,就表示Caffe CPU+Python版本环境安装成功,恭喜!

Step7:在Mnist运行Lenet,第一个CNN的试炼!

获取数据源

./data/mnist/get_mnist.sh
./examples/mnist/create_mnist.sh

修改在examples文件下的Mnist下的lenet_solver.prototxt中的solver_mode:CPU

sudo gedit ./examples/mnist/lenet_solver.prototxt
solver_mode: CPU

训练模型

./examples/mnist/train_lenet.sh

Caffe初学者第一部:Ubuntu14.04上安装caffe(CPU)+Python的详细过程 (亲测成功, 20180524更新)

Caffe初学者第一部:Ubuntu14.04上安装caffe(CPU)+Python的详细过程 (亲测成功, 20180524更新)

等待训练完成。至此,本教程结束。

后续:

本教程亲测成功,如有问题欢迎留言讨论。

本教程只针对初学者安装最简单的Caffe(CPU)+Python环境

本博主会再写后续Caffe初学者四部曲博客,尽请期待:

Caffe初学者第一部:Caffe(CPU)+Python教程

Caffe初学者第二部:Caffe(CPU)+Matlab教程

Caffe初学者第三部:Caffe(GPU)+Python教程

Caffe初学者第四部:Caffe(GPU)+Matlab教程

参考博客:

http://blog.csdn.net/wopawn/article/details/52302164

https://www.jianshu.com/p/dfd9436956b9

https://www.cnblogs.com/xuanxufeng/p/6016945.html

https://blog.csdn.net/u014696921/article/details/52156552