1. はじめに
- ディープラーニング用のマシンを複数組んだので、マニュアル代わりに
2. 要件
- tensorflow-gpuをインストールのみ!!
3. 各種インストール
(1) cudaのインストール
- https://developer.nvidia.com/cuda-80-ga2-download-archive からinstallerをインストール(cuda-8.0をインストールする)
- 以下のコマンドでcudaのインストール
$ sudo dpkg -i cuda-repo-ubuntu1604_8.0(tabで補完)
$ sudo apt-get update
$ sudo apt-get install cuda
3. パスを追加する
# 以下を.bashrc or .bash_profile or .zshrcなどに追加する
# ----------
export PATH=/usr/local/cuda-8.0/bin:${PATH}
export LD_LIBRARY_PATH=/usr/local/cuda-8.0/lib64:${LD_LIBRARY_PATH}
# ----------
# 追加後、設定ファイルの再読み込み
$ source ~/.bashrc
4. 動作確認
$ nvcc -V
# 出力例)Cuda compilation tools, release 8.0, V8.0.61
(2) tensorflow-gpuのインストール
- 基本的にgoogleのdocumentに従って行えばOK