目次
TensorFlow 1系は使える全てのGPUリソースを確保する。
メモリオーバーを観測できないので、 逐次的にGPUメモリを確保するように設定を変更する。
これによってGPU使用量の観測ができる。
ただし実行速度は劣化する。
記述例
config = tf.ConfigProto(
gpu_options=tf.GPUOptions(
allow_growth=True
)
)
sess = sess = tf.Session(config=config)
機械学習の基礎からモデルの仕組みまで体系的に学びたい方へ
参考リンク
関連書籍