Ubuntu 16.04.3にCUDA 8.0とcuDNN 6.0をインストールして、TensorFlow GPUの環境を作成する

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