在tensorflow中训练小模型（<100 mb）时，GPU内存不足

Question

在tensorflow中训练小模型（<100 mb）总是使用太多GPU内存，为什么？如何理解？

Answer 1

默认情况下，tensorflow可以增长到所有GPU内存的总和。 您可以通过为会话设置配置来限制它：

gpu_memory_fraction = 0.4 # Fraction of GPU memory to use
config = tf.ConfigProto(gpu_options=
                       tf.GPUOptions(per_process_gpu_memory_fraction=gpu_memory_fraction))                          
sess = tf.Session(config=config)