firefly3399安装caffe2

 因为在firefly3399上要进行人脸识别,而模型使用的是caffe2,所以这里记录一下安装的过程.

一.安装

1.更新源,安装相关依赖

sudo apt-get update
sudo apt-get install -y --no-install-recommends \
      build-essential \
      cmake \
      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                          
sudo pip install \
      future \
      numpy \
      protobuf

2.安装共享动态库 

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

3.下载caffe2的源码,因为现在caffe被并在了pytorch里,所以我们git的是pytorch的资源

    git clone --recursive https://github.com/pytorch/pytorch.git   

4.进入pytorch目录

    cd pytorch
  git submodule update  --init

5.进行cmake,cmake 的时候最好加上sudo,不然容易报错    

 sudo cmake ..

6.编译链接安装caffe2

 sudo make install


二.测试

现在caffe2就已经安装好了,现在来测试caffe2是否安装成功

扫描二维码关注公众号,回复: 1925178 查看本文章
cd ~ 
python -c  'from caffe2.python import core' 2>/dev/null && echo "Success" || echo "Failure"


如果是"Failure",可以去这个链接查看原因:https://caffe2.ai/docs/faq.html#why-do-i-get-import-errors-in-python-when-i-try-to-use-caffe2如果返回"Success",那么说明安装已经成功了.

博主当时安装了但是返回是Failure,查看python2.7的dist-packages发现什么都没有,但是我确实安装了的呀.查找原因才知道原来是因为之前安装了Tensorflow,而firefly3399安装Tensorflow的时候会把pip改为pip3,所以上面pip其实是pip3在进行安装,如果想在python2中进行安装,那么把上面的pip改为pip2就可以了.


猜你喜欢

转载自blog.csdn.net/qq_25680531/article/details/80909300