一、依賴庫的安裝首先

1.首先需要安裝Ｕbuntu 16.04與14.04都需要的依賴庫。

sudo apt update
sudo apt 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

Jetbrains全家桶1年46，售后保障穩定

２．分版本安裝依賴庫

若系統為Ubuntu 16.04時，需要安裝libgflags-dev

sudo apt install -y --no-install-recommends libgflags-dev

若系統為Ubuntu 14.04時，需要安裝libgflags2

sudo apt-get install -y --no-install-recommends libgflags2

3.安裝一些可選的相關性

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 -H pip install --upgrade pip 

4.更新一下pip

sudo pip install \       flask \       graphviz \       hypothesis \       jupyter \       matplotlib \       pydot python-nvd3 \       pyyaml \       requests \       scikit-image \       scipy \       setuptools \       tornado

二、Caffe2安裝

首先從網上下載Caffe2安裝庫

git clone --recursive https://github.com/caffe2/caffe2.git

下載完以后進行安裝

cd caffe2
mkdir build
make -j
cd build
sudo make install

安裝結束以后我們可以使用下面的程序檢測安裝十分偶成功

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

接下來我們檢測GPU是否安裝成功，如果沒有安裝GPU，會顯示warning說是CPU only，最后test pass。這樣說明安裝成功

python -m caffe2.python.operator_test.relu_op_test

環境變量設置

首先進入環境變量

sudo gedit ~/.bashrc

在文檔的最后添加

export PYTHONPATH=/usr/local:$PYTHONPATH
export PYTHONPATH=$PYTHONPATH:/home/lemon(user)/caffe2/build
export LD_LIBRARY_PATH=/usr/local/lib:$LD_LIBRARY_PATH

添加完以后更新一下環境變量

source ~/.bashrc

到此我們安裝就已經結束了，可以去python中

import caffe2

只要沒有報錯我們就大功告成了。

安裝過程中出現到問題

1.在caffe2文件夾直接進行編譯時，出現內存不足問題

c++: internal compiler error: Killed (program cc1plus)
Please submit a full bug report,
with preprocessed source if appropriate.
See <file:///usr/share/doc/gcc-5/README.Bugs> for instructions.
caffe2/CMakeFiles/caffe2.dir/build.make:6134: recipe for target 'caffe2/CMakeFiles/caffe2.dir/queue/rebatching_queue_ops.cc.o' failed
make[3]: *** [caffe2/CMakeFiles/caffe2.dir/queue/rebatching_queue_ops.cc.o] Error 4
make[3]: *** Waiting for unfinished jobs....

在網上查詢以后發現是內存不足，我使用的是E5 2695 V3，內存32G，28線程全開時內存不足，把線程縮小到24線程時可以編譯成功。

make -j24

2.在caffe2進行編譯的時候沒有找到 CUDA 和 cuDNN 路徑

這個我們在進行編譯之前cmake一下，提前設置這兩個的路徑，在caffe2目錄下

cmake \
  -DCUDA_TOOLKIT_ROOT_DIR=/path/to/cuda/toolkit/dir \   -DCUDNN_ROOT_DIR=/path/to/cudnn/root/dir

例如我的設置為

cmake \
  -DCUDA_TOOLKIT_ROOT_DIR=/usr/local/cuda-8.0 \   -DCUDNN_ROOT_DIR=/usr/local/cuda

3.python -c ‘from caffe2.python import core’ 2>/dev/null && echo “Success” || echo “Failure”出現Failure

整個安裝過程中沒有出現問題，但是在驗證的時候就會出現錯誤，環境變量也是安裝教程所設置的，始終沒有找到原因，重新安裝折騰好幾次。最后才發現，在終端中輸入命令使其輸出環境變量

echo $PYTHONPATH

經過查看才知道，Ubuntu中環境變量法并不是安裝文檔的前后順序來進行排列的，是根據你修改的順序，后來修改的環境變量會在前邊。
出現種種問題時，先將其他環境變量注釋即可。

4.Command “python setup.py egg_info” failed with error code 1 in /tmp/pip-build-lak8RD/future/

在進行

sudo pip install \       future \       numpy \       protobuf

安裝時，出現了這個錯誤，按照提示運行下邊的命令

pip install --upgrade pip

這樣還不可以還需要安裝setuptools

sudo pip install setuptools

這樣再接著安裝命令就可以順利通過了。

參考：

caffe2 安裝教程

總結

以上是生活随笔為你收集整理的Caffe2安装_caffe教程的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。