我想弄清楚我的Tensorflow模型需要多少GPU内存。因此我使用了tf.contrib.memory_stats.MaxBytesInUse，它返回了6168 MB。
但是使用config.gpu_options.per_process_gpu_memory_fraction，我可以使用GPU的较小部分，并且模型仍然可以正常运行，而无需花费更多时间进行一个推理步骤。
是否可以确定Tensorflow模型需要多少GPU内存？我可以减少GPU的内存份额，直到TF崩溃为止，但是我猜有更优雅，更精确的方法吗？