我的本機(jī)環(huán)境如下，任何的環(huán)境上的不一致可能會(huì)帶來(lái)一些安裝上的問(wèn)題，所以這個(gè)教程只是一個(gè)簡(jiǎn)單的參考。

環(huán)境

• 操作系統(tǒng): Ubuntu 16.04
• GPU型號(hào): 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驅(qū)動(dòng)，除此以外還有CUDA和CUDNN的依賴。詳細(xì)方法可以參照這個(gè)博客的內(nèi)容。?caffe安裝教程?以下介紹一種簡(jiǎn)單的方法，根據(jù)自己環(huán)境調(diào)整

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的安裝，版本號(hào)為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自身依賴的庫(kù)。這部分可選，根據(jù)庫(kù)名可以看出這里面包含opencv這樣的圖像處理的庫(kù)，leveldb這樣的文件格式處理庫(kù)，openmpi這樣的mpi通訊框架（用于分布式訓(xùn)練）等等，以及一些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項(xiàng)目并且編譯，clone的速度可能不會(huì)很快，因?yàn)檫€需要下載一些第三方的依賴文件，花費(fèi)時(shí)間比較長(zhǎng)。

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"

上面給出了編譯之后測(cè)試是否成功的方法，在python中導(dǎo)入相應(yīng)的包看看是否成功。

python -m caffe2.python.operator_test.relu_op_test

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

vim ~/.bashrc #在文件的最后一行插入以下內(nèi)容(需要替換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

總結(jié)

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

如果覺(jué)得生活随笔網(wǎng)站內(nèi)容還不錯(cuò)，歡迎將生活随笔推薦給好友。