在Tensorflow中测量详细GPU内存使用率的最佳方法是什么

import tensorflow as tf
from tensorflow.python.client import device_lib

local_device_protos = device_lib.list_local_devices()
gpu_devices = [x for x in local_device_protos if x.device_type == 'GPU']

gpu_device = gpu_devices[0]  # 选择第一个GPU设备

config = tf.ConfigProto()
config.gpu_options.allow_growth = True  # 允许GPU内存按需增长
config.gpu_options.visible_device_list = str(gpu_device.physical_device_desc)  # 指定要使用的GPU设备
sess = tf.Session(config=config)

gpu_memory = tf.contrib.memory_stats.MaxBytesInUse()
gpu_memory_usage = sess.run(gpu_memory)

print("GPU Memory Usage: {} bytes".format(gpu_memory_usage))