ubuntu14.04+caffe2

最新推荐文章于 2019-07-21 20:03:45 发布

原创最新推荐文章于 2019-07-21 20:03:45 发布 · 1k 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#ubuntu #caffe2

操作实战专栏收录该内容

26 篇文章

订阅专栏

本文详细介绍了如何在Python2环境下安装Caffe2的步骤。首先需要安装一系列依赖库，包括cmake、protobuf等，并确保使用正确的命令进行安装。之后通过git clone下载Caffe2源码并编译安装。最后配置环境变量完成整个安装过程。

注意，目前caffe2只支持Python2，所以安装caffe2的时候需要在电脑上创建一个Python2的虚拟环境。然后安装opencv，再安装caffe2.

现在开始说caffe2的步骤：

1：安装依赖库：

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

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

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

2.下载caffe2 。注意一定要使用clone指令，不能直接下载

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"

python -m caffe2.python.operator_test.relu_op_test

3.设置环境变量：

```bash
echo $PYTHONPATH
# export PYTHONPATH=/usr/local:$PYTHONPATH
# export PYTHONPATH=$PYTHONPATH:/root/caffe2/build
echo $LD_LIBRARY_PATH
# export LD_LIBRARY_PATH=/usr/local/lib:$LD_LIBRARY_PATH
```