Caffe2 Ubuntu编译过程

最新推荐文章于 2022-05-12 14:11:22 发布

浪Q拜

最新推荐文章于 2022-05-12 14:11:22 发布

阅读量392

点赞数

CC 4.0 BY-SA版权

分类专栏： c/c++ caffe2

本文链接：https://blog.youkuaiyun.com/u013108408/article/details/85697164

c/c++ 同时被 2 个专栏收录

4 篇文章

订阅专栏

caffe2

4 篇文章

订阅专栏

Caffe2 Ubuntu编译过程

环境准备

Vmware workstation安装
Ubuntu系统安装

编译流程

导入依赖库

sudo apt-get update
sudo apt-get install -y --no-install-recommends \
      build-essential \
      git \
      libgoogle-glog-dev \
      libgtest-dev \
      libiomp-dev \
      libleveldb-dev \
      liblmdb-dev \
      libopencv-dev \
      libopenmpi-dev \
      libsnappy-dev \
      libprotobuf-dev \
      openmpi-bin \
      openmpi-doc \
      protobuf-compiler \
      python-dev \
      python-pip                          
pip install --user \
      future \
      numpy \
      protobuf \
      typing \
      hypothesis

安装gflags、CMAKE

sudo apt-get install -y --no-install-recommends \
      libgflags-dev \
      cmake

由于系统环境为虚拟机环境，没有对GPU进行虚拟化，此处省略相关的GPU步骤，若需要支持GPU请参见官网执行操作

获取源码

git clone https://github.com/pytorch/pytorch.git && cd pytorch
git submodule update --init --recursive

编译Caffe2源码

如果使用官网的操作

python setup.py install

有时候会出现setup.py未找到等问题，即使找到了执行了编译过程，编译到最后无法进行文件夹的创建，出现Permission Denied问题
此处需执行
cd 源码所在文件夹
比如，


 cd pytorch

然后执行

sudo python setup.py install

等待编译完成

编译验证

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

结果为Success则编译成功，否则查看错误日志改正后继续编译