轉載請宣告，謝謝。

• • 特別注意
  • 安裝依賴
  • 下載原始碼
  • 編譯和安裝
  • 測試
  • 環境配置

特別注意

目前caffe2 Github更新頻繁，經常出現build error狀態，建議在如下狀態進行git操作。

安裝依賴

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支援 (可選)

如果打算使用GPU訓練模型，需要安裝CUDA和CUDNN。
需要安裝N卡閉源驅動！
在ubuntu16.04上安裝CUDA：

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

安裝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/local
rm 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-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

原始碼較大，加上子模組有160兆。

編譯和安裝

cd caffe2
make && 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:$PYTHONPATH
export PYTHONPATH=$PYTHONPATH:/path/to/caffe2/build
export LD_LIBRARY_PATH=/usr/local/lib:$LD_LIBRARY_PATH

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

至此caffe2安裝完畢。