tensorflow 如何知道实际使用gpu大小

在使用TensorFlow时，可以使用tf.config.list_physical_devices('GPU')函数来获得系统中可用的GPU设备列表。然后，可以使用tf.config.experimental.get_memory_usage(<device>)函数来获取特定GPU设备的内存使用情况。

以下是一个示例代码，演示如何获取实际使用的GPU大小：

import tensorflow as tf

# 获取可用的GPU设备列表
gpus = tf.config.list_physical_devices('GPU')
if gpus:
    for gpu in gpus:
        # 获取GPU设备的内存使用情况
        mem_usage = tf.config.experimental.get_memory_usage(gpu)
        print(f"GPU memory usage: {mem_usage.current_bytes / 1024**3} GB / {mem_usage.total_bytes / 1024**3} GB")
else:
    print("No GPU devices found.")

这将输出每个可用GPU设备的内存使用情况，包括当前使用的字节数和总字节数