[設定] tensorflowメモリオーバヘッド

問題

あなたはTFを実行すると、一般的に、単一のプログラムは、多くの場合、GPUのすべてのメモリ・ブロックを使用しています。
しかし、実際には、プログラムは、おそらくこのような大きなメモリ空間を必要としません。

改善

tf.ConfigProto().gpu_options.allow_growth=Trueプログラムを伝えるだけに必要な実際のメモリを取ります：

# ---------------- session ----------------
config = tf.ConfigProto()
config.gpu_options.allow_growth = True
sess = tf.Session(config=config)

ここでの場合はconfig.gpu_options.allow_growthセットFalse、プログラムは、すべてのメモリの塗りつぶしカード全体にデフォルト設定されます。