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/
    • さあ、早速使ってみましょう。ここをクリック。
    • 終わり
misc
スポンサーリンク
シェアする
demura.netをフォローする
demura.net

コメント

タイトルとURLをコピーしました