Caffe安装教程之Ubuntu16.04

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

要进行深度学习的研究工作,就不得不提到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)。

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


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

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

[plain] view plain copy
  1. sudo apt-get install libatlas-base-dev  

        4.安装剩余依赖:

[plain] view plain copy
  1. sudo apt-get install libgflags-dev libgoogle-glog-dev liblmdb-dev  

        5.下载Caffe:

[plain] view plain copy
  1. sudo apt-get install git  
  2. git clone https://github.com/BVLC/caffe.git  

        6.修改Makefile.config:

[plain] view plain copy
  1. cp Makefile.config.example Makefile.config  
  2. gedit Makefile.config  

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

        找到

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

        8.耐心等待编译完成

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