tensorflowとcupyのwindows環境インストール（cuDNN、CUDA）（Linuxのみ2018/1/21時点の情報）

追記：WindowsはCUDA9.1未対応。Windowsは下記参照

tatabox.hatenablog.com

cuDNN7でpython3.6とCUDA9.1に対応。インストールが簡単になった。

1. CUDAを下記から落としてインストール

CUDA （現時点では9.1が最新）

https://developer.nvidia.com/cuda-downloads

2. cuDNNを下記から落とす。

(7以上のwindows版を選択。CUDAのバージョンに揃える)
cuDNN

https://developer.nvidia.com/rdp/form/cudnn-download-survey

3. cuDNNを展開し下記フォルダに入れる

（末尾の数値はバージョンに合わせて適宜変更）
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.1

4. 下記サイトでVisual C++ Build Tools 2015をインストール

Download the Visual C++ Build Tools (standalone C++ compiler, libraries and tools)

cuDNNはVS2017には対応していないので注意

5. 環境変数に下記を設定

(CUDAの後の番号はバージョンに応じて適宜変更）
INCLUDE
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.1\include
C:\Program Files (x86)\Windows Kits\10\Include\10.0.10240.0\ucrt
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.1\bin
PATH
C:\Program Files (x86)\Microsoft Visual Studio 14.0\VC\bin

6. pip install

pip install cupy
pip install tensorflow-gpu