Kerasの一部のライブラリに対して、GPUが計算に使われない。

Question

VGG16を転移学習させて「まどか☆マギカ」のキャラを見分ける
上のリンク先のサイトのプログラムを回してみたところ、
以下の箇所(#プログラム1)でGPUを使って計算してくれなかったので、計算時間が結構かかってしまいました。

実行環境は以下のようになります。
・実行環境
GPU:RTX2060
CUDA Toolkit 10.0.130
cuDNN v7.6.5 (November 5th, 2019), for CUDA 10.0
tensorflow-gpu 2.0.0
keras 2.3.1
Python3.6.10(Python3.7だとできないらしい)
jupiter notebook 6.0.3
anaconda3
windows10 64bit

Answer 1

推測ですが、ImageDataGeneratorなどのgenerator部分に時間がかかってしまい、大部分がCPUで画像を作成している時間になってしまい、使われていないという表示になってしまうという可能性があります。

プログラムによって使われる使われない分かれるのは絶対にないので（ライブラリが違う以外）一度batch_sizeをメモリが許す限り大きい数にして学習させてみてください。質問者さんの環境を見ますと、RTX2060ということは相当早いので本当に一瞬で終わってしまうのではないでしょうか。