【手順メモ】Windows10にTensorflow gpu 2.0をインストール【自分用】

概要

Build from source on Windows  |  TensorFlow
ここを参考にバージョンをあわせて，以下を実行．

1. pipによるTensorflow 2.0のインストール
2. CUDAのインストール
3. CuDNNのインストール
4. PATHを通す

ちなみにPythonのバージョンは，3.6.8です．

Tensorflow2.0のインストール

pip install tensorflow-gpu==2.0.0

待つ！以上！

別バージョンが既に入っていたらアンインストール

pip uninstall tensorflow-gpu

CUDAのインストール

CUDA Toolkit Archive | NVIDIA Developer
ここから『CUDA Toolkit 10.0 (Sept 2018)』を選択し，
Windows -> x86_64 -> 10 -> exe(local) でダウンロード．

ダウンロード後，exeファイルを実行し，次へ進む連打でインストール．

この段階で，C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0が生成されている（はず）．

cuDNN

cuDNN Archive | NVIDIA Developer
ここから『Download cuDNN v7.4.1 (Nov 8, 2018), for CUDA 10.0』をダウンロード．
解凍したら，中身のbin, include, libを先程のフォルダ(C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0)へ上書き保存．

PATHを通す

コントロールパネル -> システム -> システムの詳細設定 -> 環境変数 -> "Path"を編集
以下を追加

• C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0\bin
• C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0\extras\CUPTI\libx64
• C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0\include
• C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0\libnvvp

正しく実行できるか確認

import tensorflow as tf
print(tf.__version__) # 2.0.0