![見出し画像](https://assets.st-note.com/production/uploads/images/45360168/rectangle_large_type_2_3a7d2b0049c4866b41133b407eb96070.png?width=1200)
Photo by
hamahouse
tensorflow GPUメモリを一気に確保しない設定
tensorflow 1系は使える全てのGPUリソースを確保する。
メモリオーバーを観測できないので、 逐次的にGPUメモリを確保するように設定を変更する。
これによってGPU使用量の観測ができる。
ただし実行速度は劣化する。
記述例
config = tf.ConfigProto(
gpu_options=tf.GPUOptions(
allow_growth=True
)
)
sess = sess = tf.Session(config=config)
参考リンク
私にカフェオレを飲ませるためにサポートしてみませんか?