ubuntu16.04安装caffe2(CPU only)

前言

前段时间贾清扬大牛在Facebook推出了caffe2版本,相比caffe来说感觉更加轻量级和高效。

Caffe2的安装相比于caffe在安装的时候更加简便,略去了Makefile.config的各种配置,对于有无GPU以及各种可选库例如opencv,anaconda的支持也更简单。(其实你直接装好库以后make就好,以GPU为例,在make的时候,自动检测你是否安装了CUDA,若没有,就自动CPU only)

在开始安装之前,附上caffe2的官方链接,官网对于安装还是讲的非常详细易懂。博主底下会给出安装之后具体截图,方便大家理解,当然对于大神们可以直接点链接然后略过后文了。XD

https://caffe2.ai/docs/getting-started.html?platform=ubuntu&configuration=compile

一、依赖库的安装

装过caffe的这个应该都比较清楚

sudo apt update
sudo apt 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

若版本为Ubuntu16.04,则安装libgflags-dev

sudo apt install -y --no-install-recommends libgflags-dev

若版本为Ubuntu14.04,则安装libgflags2

sudo apt-get install -y --no-install-recommends libgflags2

这两个不能搞混,否则安装会报错

其他的一些optional dependencies(ubuntu16.04与14.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 -H pip install --upgrade pip 
##这里更新一下pip,否则后面pip安装jupyter有可能会报错

sudo pip install \
      flask \
      graphviz \
      hypothesis \
      jupyter \
      matplotlib \
      pydot python-nvd3 \
      pyyaml \
      requests \
      scikit-image \
      scipy \
      setuptools \
      tornado

博主的习惯其实就是把上面的所有写进一个.sh文件中,例如dependencies.sh
然后执行sh dependencies.sh 这样简单省事

二、安装

首先是git

git clone --recursive https://github.com/caffe2/caffe2.git
cd caffe2
make

caffe2的好处就是makefile中会根据你的库进行make,省去了各种makefile.config的设置,不像caffe有的时候真的是会被CUDA还有各种库折腾很久。

这里写图片描述
make成功以后图如下:
这里写图片描述

之后:

cd build
sudo make install

这里写图片描述
如图所示,安装成功

可以用下面这行import python的东东:

python -c 'from caffe2.python import core' 2>/dev/null && echo "Success" || echo "Failure"

这里写图片描述
会echo “success”

最后可以用

python -m caffe2.python.operator_test.relu_op_test

来检测你的GPU是否安装成功,如果没有安装GPU,他会给以个warning说是CPU only,最后test pass。这样说明我们安装成功(因为本身我们安装的就是CPU only版本)
这里写图片描述
//warning//
这里写图片描述
//run test ok//

三、环境变量设置

几行代码OK

echo $PYTHONPATH
# export PYTHONPATH=/usr/local:$PYTHONPATH
# export PYTHONPATH=$PYTHONPATH:/home/pengyu/caffe2/build ##这里大家别忘记该路径
echo $LD_LIBRARY_PATH
# export LD_LIBRARY_PATH=/usr/local/lib:$LD_LIBRARY_PATH

#其他

可以参考官网,最后给出的关于notebook的配置,需要的人可以按照其配置。
官网也给出了如果安装出错的相应原因,大家可以去浏览。

你可能感兴趣的:(深度学习,ubuntu,caffe2)