前に逃げる

理系の大学院生。ディープラーニング初心者。

JetsonTX1にcaffe2をビルド

Jetson TX1にcaffe2をインストール

f:id:dl-kento:20180204230929p:plain

Jetsonで物体検出とかやりたい.庭に遊びに来る猫とか鳥とかを自動で撮影できたらいいな. 手順は公式HPのと大体同じです.

依存ライブラリをインストール

Caffe2の公式ページを参考に依存ライブラリをインストール

sudo apt-get update
sudo apt-get install -y --no-install-recommends \
      build-essential \
      cmake \
      git \
      libgoogle-glog-dev \
      libgtest-dev \
      libiomp-dev \
      libleveldb-dev \
      liblmdb-dev \
      libopencv-dev \
      libopenmpi-dev \
      libsnappy-dev \
      libprotobuf-dev \
      openmpi-bin \
      openmpi-doc \
      protobuf-compiler \
      python-dev \
      python-pip

sudo pip install \
      future \
      numpy \
      protobuf

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

しかし、libopencv-devが入らなかったのでスキップ。 OpenCVJetPackインストール時に入るようなので大丈夫っぽい。

Caffe2のリポジトリをクローン&ビルド

# リポジトリのクローン
git clone --recursive https://github.com/caffe2/caffe2.git && cd caffe2

# cmake
mkdir build && cd build
cmake ..

# make
sudo make install

ビルドは結構時間かかりました.

環境変数

.bashrcにパスを通しておきました.

export PYTHONPATH=/usr/local:$PYTHONPATH
export PYTHONPATH=$PYTHONPATH:/home/ubuntu/caffe2/build
export LD_LIBRARY_PATH=/usr/local/lib:$LD_LIBRARY_PATH

動作確認

python -m caffe2.python.operator_test.relu_op_test