欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

Caffe学习1——在ubuntu16.04环境下安装Caffe(CPU)

程序员文章站 2022-07-14 12:38:42
...

目前在网上,已经存在很多的Caffe安装教程。本文主要介绍笔者在安装Caffe——CPU版本所遇到的问题(主要参考了网上的  博客) ,以及相关解决方法。以下给出基于ubuntu 16.04系统的Caffe的CPU安装步骤:

1. 安装git

sudo apt-get install git

2. 安装依赖库

sudo apt-get install libprotobuf-dev 
sudo apt-get install libleveldb-dev
sudo apt-get install libsnappy-dev 
sudo apt-get install libopencv-dev
sudo apt-get install libhdf5-serial-dev
sudo apt-get install protobuf-compiler
sudo apt-get install libgflags-dev
sudo apt-get install libgoogle-glog-dev
sudo apt-get install liblmdb-dev
sudo apt-get install libatlas-base-dev

3. 从github上下载Caffe源代码

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

在此步,笔者git clone出现过两个问题:

  • 第一个问题,是笔者在公司下载Caffe时,会出现
  • 第二个问题,是笔者是在华为云下载Caffe,由于笔者购买的是固定宽带1M,导致网速受限,一直下载失败,建议同样用云服务器的同学可以找一个空闲的时间来进行下载。

4. 进入Caffe文件夹开始安装

cd caffe
cp Makefile.config.example Makefile.config

此后将Makefile.config中的语句进行修改。

  • 去掉下面语句中的注释#号,即让本次编译设定为只编译CPU版本。
# CPU_ONLY := 1
  • 添加头文件和库文件的相关路径
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 /usr/lib/x86_64-linux-gnu/

即找到INCLUDE_DIRS和LIBRARY_DIRS关键字,分别添加成上面的字段。

  • 编译文件
sudo make all
sudo make test
sudo make runtest

此处笔者遇到了一定的问题,编译make all时,首先是缺少boost,那就按照boost

sudo apt-get install libboost-dev

安装boost之后,重新编译又会遇到无法找到boost_system、boost_filesystem和 boost_thread。此时也可以分别对这三个进行安装,笔者是这么做的

sudo apt-get install libboost-system-dev
sudo apt-get install libboost-filesystem-dev
sudo apt-get install libboost-thread-dev

后来才发现,原来可以直接全部安装(还是自己蠢了)

sudo apt-get install libboost-all-dev

安装之后,就可以进行编译安装了。

5. 编译Python接口

此时已经可以使用编译完成的.a或者.so文件来训练网络了,但现阶段主要使用的Python接口,而笔者在编译Python接口的时候,主要参考了caffe安装教程的Python接口安装过程,并未遇到问题,这里就不再进行阐述了。

6. 参考

  • https://blog.csdn.net/u010193446/article/details/53259294