Ubuntu16.04: Caffe-0.15、tensorflow-1.4.1とDigits6.0のインストール

Ubuntu16.04にCaffe-0.15、TensorFlow-1.4.1とDIGITS6.0を入れたときのメモ

Caffe-0.15のインストール
- 準備：Protbuf3をソースからインストール。バイナリだとDigits内部でTensorflowがうまく動かないらしい。
  - sudo apt-get install autoconf automake libtool curl make g++ git python-dev python-setuptools unzip
  - ~/.bashrcの最後へ以下を追加して保存
    - export PROTOBUF_ROOT=~/src/protobuf
  - cd
  - source ~/.bashrc
  - cd src
  - git clone https://github.com/google/protobuf.git $PROTOBUF_ROOT -b '3.2.x'
  - cd $PROTOBUF_ROOT
  - ./autogen.sh
  - ./configure
  - make “-j$(nproc)”
  - sudo make install
  - sudo ldconfig
  - cd python
  - sudo python setup.py install --cpp_implementation
- 依存関係
  - sudo apt-get install –no-install-recommends build-essential cmake git gfortran libatlas-base-dev libboost-filesystem-dev libboost-python-dev libboost-system-dev libboost-thread-dev libgflags-dev libgoogle-glog-dev libhdf5-serial-dev libleveldb-dev liblmdb-dev libopencv-dev libsnappy-dev python-all-dev python-dev python-h5py python-matplotlib python-numpy python-opencv python-pil python-pip python-pydot python-scipy python-skimage python-sklearn
  - sudo apt-get install libgflags-dev libgoogle-glog-dev liblmdb-dev libleveldb-dev libsnappy-dev libatlas-base-dev doxygen
- ソースコードのダウンロード
  - ~/.bashrcの最後の行へ次を追加して保存する。
    - export CAFFE_ROOT=~/src/caffe
  - source ~/.bashrc
  - cd src
  - git clone https://github.com/NVIDIA/caffe.git -b caffe-0.15
- Pythonパッケージのインストール
  - sudo pip install -r $CAFFE_ROOT/python/requirements.txt
- ビルド
  - cd $CAFFE_ROOT
  - mkdir build
  - cd build
  - cmake ..
  - make -j”$(nproc)”
    - 次のようなエラーが出た場合はここを参照して/usr/include/cudnn.hのinclude “driver_types.h”を<driver_types.h>に変更する。
    - /usr/local/cuda-8.0/include/cuda_runtime_api.h:1628:101: error: use of enum ‘cudaDeviceP2PAttr’ without previous declaration
      extern __host__ __cudart_builtin__ cudaError_t CUDARTAPI cudaDeviceGetP2PAttribute(int *value, enum cudaDeviceP2PAttr attr, int srcDevice, int dstDevice);
  - make install
TensorFlow1.4.1のインストール
- sudo apt-get install libcupti-dev
- sudo apt-get install python3-pip python3-dev python-virtualenv
- mkdir ~/src/tensorflow
- virtualenv --system-site-packages -p python3 ~/src/tensorflow
- source ~/src/tensorflow/bin/activate を実行するとプロンプトが次のように変わる。
  - (tensorflow)$ easy_install -U pip
  - (tensorflow)$ pip3 install --upgrade tensorflow-gpu
  - tensorflowを終了するときは次のコマンドを実行する
  - (tensorflow)$ deactivate
DIGITSのインストール
- ダウンロードするためにはNVIDIA Developerへの登録が必要。
- 依存関係
  - sudo apt-get install --no-install-recommends git graphviz python-dev python-flask python-flaskext.wtf python-gevent python-h5py python-numpy python-pil python-pip python-scipy python-tk
- ソースのダウンロード
  - ~/.bashrcの最後の行へ次を追加して保存する。
    - export DIGITS_ROOT=~/src/digits
  - source ~/.bashrc
  - cd ~/src
  - git clone https://github.com/NVIDIA/DIGITS.git
- Pythonパッケージ
  - sudo pip install -r $DIGITS_ROOT/requirements.txt
  - sudo pip install -e $DIGITS_ROOT
- サーバーのスタート
  - cd ~/src/digits
  - ./digits-devserver
- ウェブブラウザを開き、次のURLを入力する
  - http://localhost:5000/
- さあ、早速使ってみましょう。ここをクリック。
- 終わり