我的本機環境如下，任何的環境上的不一致可能會帶來一些安裝上的問題，所以這個教程只是一個簡單的參考。

環境

• 操作系統: Ubuntu 16.04
• GPU型號: Tesla M40 24GB
• Python: 2.7 路徑 /usr/bin/python即全局的python解釋器

caffe2必備依賴的安裝

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的用戶需要GPU驅動，除此以外還有CUDA和CUDNN的依賴。詳細方法可以參照這個博客的內容。?caffe安裝教程?以下介紹一種簡單的方法，根據自己環境調整

cuda的apt-get安裝，版本8.0

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的安裝，版本號為5.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

caffe2需要的其他依賴

同樣的，還需要一些caffe2自身依賴的庫。這部分可選，根據庫名可以看出這里面包含opencv這樣的圖像處理的庫，leveldb這樣的文件格式處理庫，openmpi這樣的mpi通訊框架（用于分布式訓練）等等，以及一些python上可選的依賴。

# 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 \future \graphviz \hypothesis \jupyter \matplotlib \pydot python-nvd3 \pyyaml \requests \scikit-image \scipy \setuptools \six \tornado

編譯安裝

clone項目并且編譯，clone的速度可能不會很快，因為還需要下載一些第三方的依賴文件，花費時間比較長。

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"

上面給出了編譯之后測試是否成功的方法，在python中導入相應的包看看是否成功。

python -m caffe2.python.operator_test.relu_op_test

執行以上的python模塊化執行的命令可以測試是否能夠正常地在caffe2中使用GPU，需要注意的是此時不能更改目錄，還需要在make時候進入的build目錄中。 為了能夠在所有目錄中都能正常使用caffe2，需要修改環境變量。建議修改用戶目錄下的.bashrc文件，這樣系統的不同用戶之間不會沖突。

vim ~/.bashrc #在文件的最后一行插入以下內容(需要替換caffe2的路徑) export PYTHONPATH=/usr/local:$PYTHONPATH export PYTHONPATH=$PYTHONPATH:/path/to/caffe2_ROOT/buildexport LD_LIBRARY_PATH=/usr/local/lib:$LD_LIBRARY_PATH#然后保存文件重新source一下 source ~/.bashrc

總結

以上是生活随笔為你收集整理的Ubuntu16.04 Caffe2 编译安装步骤记录的全部內容，希望文章能夠幫你解決所遇到的問題。

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