GPU版TensorFlowをWindowsにインストール

2017.07.24
TensorFlow

これまでずっとCPU版のTensorFlowを使っていて、それほど困らなかったのですが、
大量の画像を学習しようとすると流石にかなり遅くなる為、
GPU版のインストールを行いました。

1. 必要環境
2. TensorFlowインストール
3. 効果

必要環境

CUDA Toolkit Download | NVIDIA Developer
　特に悩むところはありません。OSとバージョンを選択してインストールします。

NVIDIA cuDNN | NVIDIA Developer
　ダウンロードするにはメンバー登録が必要な為、適当に登録しておきます。
　展開したファイルはCUDAがインストールされているフォルダに同名のフォルダがあるので、
　それぞれにコピーします。
　【※】ダウンロードするのはcuDNN v5.1です。最新の6.0ではうまく動きませんでした。

TensorFlowインストール

公式ページに書かれている通り、下記を実行して、TensorFlowをインストールします。
最新のバージョンのコマンドは公式ページで確認して下さい。

pip install --ignore-installed --upgrade https://storage.googleapis.com/tensorflow/windows/gpu/tensorflow_gpu-<span class="synConstant">1.2</span>.<span class="synConstant">1</span>-cp35-cp35m-win_amd64.whl

1	pip install --ignore-installed --upgrade https://storage.googleapis.com/tensorflow/windows/gpu/tensorflow_gpu-<span class="synConstant">1.2</span>.<span class="synConstant">1</span>-cp35-cp35m-win_amd64.whl