TensorFlow GPUの環境を作成する
- 計算機環境
$ cat /etc/lsb-release
DISTRIB_ID=Ubuntu
DISTRIB_RELEASE=16.04
DISTRIB_CODENAME=xenial
DISTRIB_DESCRIPTION="Ubuntu 16.04.3 LTS"
- Ubuntu 16.04.3にCUDA 8.0とcuDNN 6.0をインストールする
- 最新のCUDA 9.0とcuDNN 7.0はTensorFlow GPU 1.3.0 に対応していないもよう
- Upgrade to CuDNN 7 and CUDA 9 · Issue #12052 · tensorflow/tensorflow · GitHub
1. NVIDIAドライバーを入れる
- 古いドライバの削除
$ dpkg -l | grep nvidia
$ dpkg -l | grep cuda
$ sudo apt-get --purge remove nvidia-*
$ sudo apt-get --purge remove cuda-*
- NVIDIAドライバーをインストール
$ sudo add-apt-repository ppa:graphics-drivers/ppa
$ sudo apt-get update
$ apt-cache search 'nvidia-[0-9]+$'
$ sudo apt install nvidia-384
$ reboot
2. CUDA 8.0を入れる
- CUDA Toolkit Archive から CUDA Toolkit 8.0 をダウンロードする
-
installではcuda-8.0を指定する
$ sudo dpkg -i cuda-repo-ubuntu1604-8-0-local-ga2_8.0.61-1_amd64.deb
$ sudo apt-get update
$ sudo apt-get install cuda-8-0
3. cuDNN v6.0を入れる
-
cuDNN Download から
Download cuDNN v6.0 (April 27, 2017), for CUDA 8.0をダウンロードする
$ tar xfvz cudnn-8.0-linux-x64-v6.0.tgz
$ sudo cp cuda/include/cudnn.h /usr/local/cuda-8.0/include/
$ sudo cp cuda/lib64/libcudnn* /usr/local/cuda-8.0/lib64/
$ sudo chmod a+r /usr/local/cuda-8.0/lib64/libcudnn*
-
.bashrcなどでパスの設定を行う
export PATH="/usr/local/cuda/bin:$PATH"
export LD_LIBRARY_PATH="/usr/local/cuda/lib64:$LD_LIBRARY_PATH"
4. TensorFlow GPUを入れる
- インストール
$ sudo apt-get install libcupti-dev
$ sudo pip install tensorflow-gpu
- 確認
$ python -c "import tensorflow"
###（2018/01/25 追記）
最新の TensorFlow 1.4.1 に Python 3.6 向けのパッケージがないらしい
$ python -c "import tensorflow"
/usr/local/pyenv/versions/3.6.4/lib/python3.6/importlib/_bootstrap.py:219: RuntimeWarning: compiletime version 3.5
of module 'tensorflow.python.framework.fast_tensor_util' does not match runtime version 3.6
return f(*args, **kwds)
対策としては下記の通り（参考）
- 公式対応を待つ
- 警告を無視して使う
- 1.3.0 を使う
$ sudo pip install tensorflow-gpu==1.3.0