Windows10にTensorflow-GPUを導入する[2021/01/18]

基本は

初心者がGPU搭載Windows10にPython + Anaconda + TensorFlow + Kerasの環境を構築してみた[2018/4/28]

に準拠

TensorFlow公式サイトで確認をする

ただし、(恐らく公式サイトが更新していしないため、CUDA10.1用のcuDNNのバージョンが存在しないため、cuDNN 7.6を使用。以下のファイルを入手。

MSVCはVisual Studio Community 2019

cuda_10.1.243_426.00_win10.exe

cudnn-10.1-windows10-x64-v7.6.5.32.zip

各インストール・pathの設定等を行う

Anaconda promptから仮想環境の作成とtensorflow-gpu、scipy、kerasを導入。

ただし、tensorflowはpipで導入すると2.4.0が入ってしまう

(pip install tensorflow-gpu==2.3.0)

自分は試していませんが、2.4.0での実行時に

cudart64_110.dll

cudnn64_8.dll

が必要だといわれる。おそらくcuda 11.0/cuDNN 8.xにすれば導入できるのだと思われる。

import tensorflow as tf
tf.test.is_gpu_available()

True

ここからMNISTのコードを使用し、ELSA System Graph

で確認。なお、NVIDIA GeForceは対象外の模様だけれども、動いてる様子は確認。ただし、cpu実行時との比較データは未取得。