ubuntu16.04配置caffe2（仅cpu）

最新推荐文章于 2020-08-27 20:29:31 发布

王旭东-1

最新推荐文章于 2020-08-27 20:29:31 发布

阅读量2.1k

点赞数

本文链接：https://blog.youkuaiyun.com/weixin_37911283/article/details/71122988

版权

本文详细介绍了在Ubuntu环境下安装Caffe2的过程，包括所需依赖库的安装、环境变量的配置以及如何验证安装是否成功。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

一、依赖库的安装

装过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
 
 1
2
3
4
5
6
7
8
9
10
11
 
 1
2
3
4
5
6
7
8
9
10
11

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

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

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

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

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

其他的一些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
 
 1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
 
 1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28

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

二、安装

首先是Git

git clone --recursive https://github.com/caffe2/caffe2.git
 
 1
 
 1

cd caffe2
make
 
 1
2
 
 1
2

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

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

之后：

cd build
sudo make install
 
 1
2
 
 1
2

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

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

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

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

最后可以用

python -m caffe2.python.operator_test.relu_op_test
 
 1
 
 1

来检测你的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