DeepLearning
TensorFlow
ubuntu16.04

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"

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を入れる

$ 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