環境を整えようとして大苦戦したので、判明した導入方法を紹介します。
※Intel AVX2を使うのでIntelの第二世代Coreプロセッサー(Sandy Bridge)以前の古いCPUだとインストールできません。
python
3.5.x ~ 3.7.x を利用しましょう。
モジュールインストール
pip install -U pip
pip uninstall tensorflow
pip install tensorflow-gpu==1.15.3 tensorflow-estimator==1.15.1
CUDA インストール
CUDA Toolkit 10.0 をインストールしましょう。
cuDNN の展開
v7.6.5 (November 5th, 2019), for CUDA 10.0 をダウンロードしましょう。
https://developer.nvidia.com/rdp/cudnn-download
※ログインが必要なのでログイン画面の[CREATE AN ACCOUNT]からアカウント作成(無料)してください。
ダウンロードしたzipファイルを解凍し
中身をC:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0にコピーして下さい。
PATHを通しましょう
これら3つのパスを環境変数に設定して下さい。
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
以上で、Win10にてtensorflow-gpu 1.15.3が利用可能になります。
お疲れさまでした。