前往小程序,Get更优阅读体验!
立即前往
首页
学习
活动
专区
工具
TVP
发布
社区首页 >专栏 >CentOS7安装Caffe2

CentOS7安装Caffe2

作者头像
kongxx
发布2018-05-14 13:56:32
1K0
发布2018-05-14 13:56:32
举报

安装epel

代码语言:javascript
复制
$ sudo yum install epel-release

安装依赖的系统库

代码语言:javascript
复制
# $ sudo yum update
$ sudo yum install -y \
automake \
cmake3 \
gcc \
gcc-c++ \
git \
kernel-devel \
leveldb-devel \
lmdb-devel \
libtool \
protobuf-devel \
python-devel \
python-pip \
snappy-devel \
gflags-devel \
glog-devel

安装依赖的 Python 库

代码语言:javascript
复制
$ sudo pip install --upgrade pip
$ sudo pip install \
flask \
future \
graphviz \
hypothesis \
jupyter \
matplotlib \
numpy \
protobuf \
pydot \
python-nvd3 \
pyyaml \
requests \
scikit-image \
scipy \
setuptools \
six \
tornado

编译安装 caffe2

代码语言:javascript
复制
$ git clone --recursive https://github.com/caffe2/caffe2
$ cd caffe2 && mkdir build
$ cd build && cmake3 ..
$ sudo make install

测试

首先使用下面的命令来检查 caffe2 是否安装成功

代码语言:javascript
复制
$ cd build
$ python -c 'from caffe2.python import core' 2>/dev/null && echo "Success" || echo "Failure"

如果 caffe2 安装成功,上面的命令应该打印 Success。

其次使用下面的命令来运行一个测试

代码语言:javascript
复制
$ cd build
$ python -m caffe2.python.operator_test.relu_op_test
$ python -m caffe2.python.operator_test.weighted_sample_test
$ python -m caffe2.python.examples.resnet50_trainer --train_data null --use_cpu true

问题

问题一

运行 “python -m caffe2.python.operator_test.relu_op_tes” 报错

代码语言:javascript
复制
...
Traceback (most recent call last):
  File "/usr/lib64/python2.7/runpy.py", line 162, in _run_module_as_main
    "__main__", fname, loader, pkg_name)
  File "/usr/lib64/python2.7/runpy.py", line 72, in _run_code
    exec code in run_globals
  File "/home/jhadmin/Downloads/caffe2/build/caffe2/python/operator_test/relu_op_test.py", line 22, in <module>
    from hypothesis import given
  File "/usr/lib/python2.7/site-packages/hypothesis/__init__.py", line 31, in <module>
    from hypothesis.core import given, find, example, seed, reproduce_failure, \
  File "/usr/lib/python2.7/site-packages/hypothesis/core.py", line 34, in <module>
    from coverage.files import canonical_filename
ImportError: cannot import name canonical_filename

问题原因是 Python 的 coverage 版本问题,升级 coverage 库

代码语言:javascript
复制
sudo pip install coverage --upgrade

问题二

运行 “python -m caffe2.python.operator_test.relu_op_tes” 报错

代码语言:javascript
复制
AttributeError: 'module' object has no attribute 'full'

问题原因是 numpy 版本问题,升级 numpy 库

代码语言:javascript
复制
sudo pip install numpy --upgrade
本文参与 腾讯云自媒体分享计划,分享自作者个人站点/博客。
原始发表:2018.01.19 ,如有侵权请联系 cloudcommunity@tencent.com 删除

本文分享自 作者个人站点/博客 前往查看

如有侵权,请联系 cloudcommunity@tencent.com 删除。

本文参与 腾讯云自媒体分享计划  ,欢迎热爱写作的你一起参与!

评论
登录后参与评论
0 条评论
热度
最新
推荐阅读
目录
  • 安装epel
  • 安装依赖的系统库
  • 安装依赖的 Python 库
  • 编译安装 caffe2
  • 测试
  • 问题
    • 问题一
      • 问题二
      领券
      问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档