Caffe2安装踩坑日记

【背景】Facebook 开源了物体检测研究平台 Detectron。Detectron 是 Facebook AI 研究的软件系统,实现了最先进的物体检测算法,包括 Mask R-CNN。Detectron 用Python 编写实现,并由深度学习框架 Caffe2 驱动。

Caffe2安装:(官网https://caffe2.ai/)有conda安装、github源码编译安装等多种方式。

1. conda安装

conda安装的caffe2无detectron模块,需要最新的caffe2支持,因此不能用conda安装。

2. 使用源码安装

(1)pytorch源码

前不久Caffe2 开源代码并入 PyTorch,caffe2官网提供的源码编译安装自pytorch

进行到sudo make install,错误如下:

解决:建立build之前

1)将CMakeLists.txt中local protobuf改为OFF

2)终端cmake -DCAFFE2_LINK_LOCAL_PROTOBUF=OFF

(2)caffe2源码

caffe2目录下make时,错误如下:

解决:

1)将CMakeLists.txt中MPI改为OFF

2)终端cmake .. -DUSE_MPI=OFF

最后,测试,输出Success成功。

其他问题:打开python,输入import caffe后出现ImportError: No module named caffe,此时添加环境变量,在文尾输入,保存

  • 发表于:
  • 原文链接https://kuaibao.qq.com/s/20180614G17W1000?refer=cp_1026
  • 腾讯「云+社区」是腾讯内容开放平台帐号(企鹅号)传播渠道之一,根据《腾讯内容开放平台服务协议》转载发布内容。
  • 如有侵权,请联系 yunjia_community@tencent.com 删除。

扫码关注云+社区

领取腾讯云代金券