在ubuntu16.04上使用源码安装caffe2

转载请声明，谢谢。

Facebook终于开源了caffe2，参照Tutorials，完成caffe2安装。
参考：https://caffe2.ai/docs/getting-started.html?platform=linux&configuration=compile

---

特别注意

目前caffe2 Github更新频繁，经常出现build error状态，建议在如下状态进行git操作。

安装依赖

sudo apt-get updatesudo 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支持 (可选)

如果打算使用GPU训练模型，需要安装CUDA和CUDNN。
需要安装N卡闭源驱动！
在ubuntu16.04上安装CUDA：

sudo apt-get update && sudo apt-get install wget -y --no-install-recommendswget "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.debsudo apt-get updatesudo apt-get install cuda

安装CUDNN：

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/localrm cudnn-8.0-linux-x64-v5.1.tgz && sudo ldconfig

安装可选依赖

sudo apt-get install -y --no-install-recommends \      libgflags-dev \      libgtest-dev \      libiomp-dev \      libleveldb-dev \      liblmdb-dev \      libopencv-dev \      libopenmpi-dev \      libsnappy-dev \      openmpi-bin \      openmpi-doc \      python-pydotsudo 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

源码较大，加上子模块有160兆。

编译和安装

cd caffe2make && cd build && sudo make install

测试

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

上述测试输出Success即表示安装成功

python -m caffe2.python.operator_test.relu_op_test

上述测试中，如果没有GPU支持，在开头会有警告，但是不影响运行。

环境配置

最后还需要配置环境，以便在下次启动终端能够正常运行。

cd ～/gedit .bashrc

在最后加入：

export PYTHONPATH=/usr/local:$PYTHONPATHexport PYTHONPATH=$PYTHONPATH:/path/to/caffe2/buildexport LD_LIBRARY_PATH=/usr/local/lib:$LD_LIBRARY_PATH

注意修改/path/to/caffe2指向caffe2目录。

至此caffe2安装完毕。