Caffe2 - (一)Source 安装及问题解决

AIHGF

发布于 2019-02-18 10:49:36

1.4K0

发布于 2019-02-18 10:49:36

文章被收录于专栏：AIUAIAIUAI

Caffe2 - From Source 安装

1. 安装

平台 - Ubuntu14.04 - Ubuntu16.04

从Source安装

依赖项：

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

GPU 支持

需要安装 NVIDIA CUDA 8 和 cuDNN v5.1 或 v6.0，官方推荐安装方式如下：

Step1 安装显卡驱动：

Ubuntu14.04

sudo apt-get update && sudo apt-get install wget -y --no-install-recommends
wget "http://developer.download.nvidia.com/compute/cuda/repos/ubuntu1404/x86_64/cuda-repo-ubuntu1404_8.0.61-1_amd64.deb"
sudo dpkg -i cuda-repo-ubuntu1404_8.0.61-1_amd64.deb
sudo apt-get update
sudo apt-get install cuda

Ubuntu16.04

sudo apt-get update && sudo apt-get install wget -y --no-install-recommends
wget "http://developer.download.nvidia.com/compute/cuda/repos/ubuntu1604/x86_64/cuda-repo-ubuntu1604_8.0.61-1_amd64.deb"
sudo dpkg -i cuda-repo-ubuntu1604_8.0.61-1_amd64.deb
sudo apt-get update
sudo apt-get install cuda

Step2 安装cuDNN

cuDNN v5.1

CUDNN_URL="http://developer.download.nvidia.com/compute/redist/cudnn/v5.1/cudnn-8.0-linux-x64-v5.1.tgz"
wget ${CUDNN_URL}
sudo tar -xzf cudnn-8.0-linux-x64-v5.1.tgz -C /usr/local
rm cudnn-8.0-linux-x64-v5.1.tgz && sudo ldconfig

cuDNN v6.0 NVIDIA’s cuDNN download官网注册并下载，采用类似的方式安装.

其它依赖项

Ubuntu14.04 – libgflags2

# for Ubuntu 14.04
sudo apt-get install -y --no-install-recommends libgflags2

Ubuntu16.04 – libgflags-dev

# for Ubuntu 16.04
sudo apt-get install -y --no-install-recommends libgflags-dev

其它依赖项

# for both Ubuntu 14.04 and 16.04
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

编译

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"

运行命令测试GPU-Caffe2是否编译成功：

python -m caffe2.python.operator_test.relu_op_test

环境变量设置

sudo vim ~/.bashrc
# 添加对应内容
source

echo $PYTHONPATH
# export PYTHONPATH=/usr/local:$PYTHONPATH
# export PYTHONPATH=$PYTHONPATH:/home/ubuntu/caffe2/build
echo $LD_LIBRARY_PATH
# export LD_LIBRARY_PATH=/usr/local/lib:$LD_LIBRARY_PATH

2. 错误解决

1 - ImportError: No module named past.builtins

错误问题：安装完成后，运行： python -c ‘from caffe2.python import core’ 2>/dev/null && echo “Success” || echo “Failure” 出现 Failure；运行: python -m caffe2.python.operator_test.relu_op_test 出现以下错误：

Traceback (most recent call last):
  File "/usr/lib/python2.7/runpy.py", line 162, in _run_module_as_main
    "__main__", fname, loader, pkg_name)
  File "/usr/lib/python2.7/runpy.py", line 72, in _run_code
    exec code in run_globals
  File "/usr/local/caffe2/python/operator_test/relu_op_test.py", line 21, in <module>
    from caffe2.python import core
  File "/usr/local/caffe2/python/core.py", line 24, in <module>
    from past.builtins import basestring
ImportError: No module named past.builtins

解决方案：安装 future package：

sudo  pip install future

2. gmock/gmock.h

ubuntu下 google gmock 安装. 下载 googletest，并解压

./travis.sh
cmake ./CMakeLists.txt
make
sudo make install

本文参与腾讯云自媒体分享计划，分享自作者个人站点/博客。

原始发表：2017年05月18日，如有侵权请联系 cloudcommunity@tencent.com 删除

ubuntu

linux

python

打包

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

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

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

登录后参与评论

0 条评论

热度