caffe2 安装与介绍

原创 2017年04月19日 15:31:03

一早发现caffe2的较成熟的release版发布了(the first production-ready release),那么深度学习平台在之后一段时间也是会出现其与tensorflow相互竞争的局面。
从打开这个caffe2的官网就会发现,有了Facebook的支持,连界面也好看多了。不过再仔细看看,觉得又和tensorflow有一丝像,从内到外。

Caffe 2 Caffe2


Caffe2 中基本计算单元之一是 Operators。每个 Operator 包含给定适当数量和类型的输入和参数来计算输出所需的逻辑。Caffe 和 Caffe2 功能的总体差异如下图所示:

One of basic units of computation in Caffe2 are the Operators. Each operator contains the logic necessary to compute the output given the appropriate number and types of inputs and parameters. The overall difference between operators’ functionality in Caffe and Caffe2 is illustrated in the following graphic, respectively:

caffe2
看到这段话,是不是更觉得像是tensorflow了?之前layer的概念被弱化,数据与操作完全分开,不就是tensorflow里面需要定义的tf.matmultf.Variable这类吗?
其次提出的workspace概念很像是tf中的Session:

# Create the input data
data = np.random.rand(16, 100).astype(np.float32)

# Create labels for the data as integers [0, 9].
label = (np.random.rand(16) * 10).astype(np.int32)

workspace.FeedBlob("data", data)
workspace.FeedBlob("label", label)

# Create model using a model helper
m = cnn.CNNModelHelper(name="my first net")
fc_1 = m.FC("data", "fc1", dim_in=100, dim_out=10)
pred = m.Sigmoid(fc_1, "pred")
[softmax, loss] = m.SoftmaxWithLoss([pred, "label"], ["softmax", "loss"])

网络的编写也向tf靠拢了(学了点tf还是有点用的)。

最后还要说一点就是对python的支持大大增强了,当然这也是深度学习的趋势。

安装


4.18发布的版本号为v0.7.0,官网上的安装教程比较详细,也比较好操作Install

依赖库

sudo apt-get update
sudo apt-get install -y --no-install-recommends \
      build-essential \
      cmake \
      git \
      libgoogle-glog-dev \
      libprotobuf-dev \
      protobuf-compiler \
      python-dev \
      python-pip                          
sudo pip install numpy protobuf

GPU支持

这一部分主要是CUDA与cuDNN,在之前的博客中有讲到过。

可选库

# for both Ubuntu 14.04 and 16.04
sudo apt-get install -y --no-install-recommends \
      libgtest-dev \
      libiomp-dev \
      libleveldb-dev \
      liblmdb-dev \
      libopencv-dev \
      libopenmpi-dev \
      libsnappy-dev \
      openmpi-bin \
      openmpi-doc \
      python-pydot
sudo pip install \
      flask \
      graphviz \
      hypothesis \
      jupyter \
      matplotlib \
      pydot python-nvd3 \
      pyyaml \
      requests \
      scikit-image \
      scipy \
      setuptools \
      tornado
  • 针对Ubuntu 14.04:
sudo apt-get install -y --no-install-recommends libgflags2
  • 针对Ubuntu 16.04:
sudo apt-get install -y --no-install-recommends libgflags-dev

Clone & Build

git clone --recursive https://github.com/caffe2/caffe2.git && cd caffe2
make && cd build && sudo make install
python -c 'from caffe2.python import core' 2>/dev/null && echo "Success" || echo "Failure"

这一步在我这边很慢,clone的速度大概只有几十kb,而且中途也出现了错误。如果选择用zip打包下载,就会出现caffe2/third_party/目录中第三方包缺失的情况,这时还要自己手动下载。

GPU测试可以执行:

python -m caffe2.python.operator_test.relu_op_test

这步安装完成之后,会在/usr/local/caffe2/home/user/caffe2/build(你的build路径)路径生成caffe2的python文件,在/usr/local/lib路径生成库文件。

设置环境变量

设置正确的话,执行命令会有#后这样的输出

echo $PYTHONPATH
# export PYTHONPATH=/usr/local:$PYTHONPATH
# export PYTHONPATH=$PYTHONPATH:/home/ubuntu/caffe2/build
echo $LD_LIBRARY_PATH
# export LD_LIBRARY_PATH=/usr/local/lib:$LD_LIBRARY_PATH

具体设置更改对应shell的配置文件,一般来说

sudo vim /etc/profile

在最后加上

export PYTHONPATH=/usr/local:$PYTHONPATH
export PYTHONPATH=$PYTHONPATH:/home/ubuntu/caffe2/build
export LD_LIBRARY_PATH=/usr/local/lib:$LD_LIBRARY_PATH

就可以了。

总结


目前caffe2还不是很成熟,某些文档也还不齐,安装也可以不用很着急,熟悉一下操作就好。mask-rcnn不就应该就会在这上面放出源码,可以稍稍再期待一下。

版权声明:本文为博主原创文章,转载请标注出处。

相关文章推荐

caffe转caffe2踩坑经验分享

caffe转caffe2踩坑后的吐血经验分享

caffe2 和 caffe 有何不同

前几天 facebook 开源的 caffe2,让我们在深度学习框架上又多了一个选择。caffe2 宣称是轻量级、模块化和可扩展的一个框架,code once,run anywhere。作为一个老 c...

caffe2 01 caffe2 win10 vs2017编译(默认CPU版本)

01 基本环境 win10 vs2017 git python3.5.3 cmake-gui(3.7.2)caffe2 github地址:https://github.com/caffe2/...
  • longji
  • longji
  • 2017年04月24日 11:57
  • 3870

Delphi7高级应用开发随书源码

  • 2003年04月30日 00:00
  • 676KB
  • 下载

Delphi7高级应用开发随书源码

  • 2003年04月30日 00:00
  • 676KB
  • 下载

caffe2 安装教程

前段时间贾清扬大牛在Facebook推出了caffe2版本,相比caffe来说感觉更加轻量级和高效。Caffe2的安装相比于caffe在安装的时候更加简便,略去了Makefile.config的各种配...

caffe2 Windows安装和编译

caffe2地址:https://github.com/caffe2/caffe2 caffe2是caffe升级的版本,但和caffe并不兼容,不能直接替换,需要用提供的脚本进行转换。 官网给出的...

【Caffe安装】caffe安装系列——史上最详细的安装步骤

说明 网上关于caffe的安装教程非常多,但是关于每一步是否操作成功,出现了什么样的错误又该如何处理没有给出说明。因为大家的操作系统的环境千差万别,按照博客中的教程一步步的安装,最后可能失败——这是...
  • haoji007
  • haoji007
  • 2016年07月31日 22:14
  • 41375

ubuntu 16.04下安装caffe2

上一篇文章说了如何在Ubuntu 16.04安装Cuda8.0。 在Ubuntu 16.04完整安装Caffe2的话,安装顺序为:安装Cuda8.0------>安装cuDNN8.0---->源码编...

caffe2 安装

caffe2 安装过程
内容举报
返回顶部
收藏助手
不良信息举报
您举报文章:caffe2 安装与介绍
举报原因:
原因补充:

(最多只允许输入30个字)