Caffe安装教程之Ubuntu16.04

时间:2022-05-11 06:27:50

        要进行深度学习的研究工作,就不得不提到Caffe,一个优秀的深度学习框架。

        那我们应该如何安装Caffe呢?其实Caffe的官网上提供了非常好的安装教程,但所对应的Ubuntu版本为14.04,12.04,没有16.04版的安装教程(大哭),本篇博文就是教大家在Ubuntu16.04上安装Caffe的。

        那我们要准备什么呢?所需软件:Oracle VM VirtualBox,Ubuntu16.04。

        1.首先我们需要在VirtualBox中安装Ubuntu16.04,傻瓜式操作。

        2.系统安装完成后,打开命令行控制台,输入以下命令,安装基本依赖(General dependencies)。

sudo apt-get install libprotobuf-dev libleveldb-dev libsnappy-dev libopencv-dev libhdf5-serial-dev protobuf-compiler
sudo apt-get install --no-install-recommends libboost-all-dev

        注意:由于网络情况的不同,一次安装可能不能全部安装成功,重复输入上述命令,再次安装(本人用了两次),这里建议大家一次安装一个软件包,便于弄清是那个未安装成功,以节省时间。

        3.安装ATLAS,输入下述命令:

sudo apt-get install libatlas-base-dev

        4.安装剩余依赖:

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

        5.下载Caffe:

sudo apt-get install git
git clone https://github.com/BVLC/caffe.git

        6.修改Makefile.config:

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

        找到#CPU_ONLY := 1,取消注释(我们设置为CPU模式)

        找到

# Whatever else you find you need goes here.
INCLUDE_DIRS := $(PYTHON_INCLUDE) /usr/local/include
LIBRARY_DIRS := $(PYTHON_LIB) /usr/local/lib /usr/lib
        修改为
# Whatever else you find you need goes here.INCLUDE_DIRS := $(PYTHON_INCLUDE) /usr/local/include /usr/include/hdf5/serialLIBRARY_DIRS := $(PYTHON_LIB) /usr/local/lib /usr/lib /usr/lib/i386-linux-gnu/hdf5/serial
        7.进行编译
make all
make test
make runtest

        8.耐心等待编译完成

        好了,到此为止,Caffe就安装完成了。尽情使用吧!