import tensorflow as tf gpus = tf.config.experimental.list_physical_devices(device_type='GPU') for gpu in gpus: tf.config.experimental.set_memory_growth(gpu, True)

这段代码用于在 TensorFlow 中设置 GPU 内存动态增长。首先，它导入了 TensorFlow 库。然后，使用 `list_physical_devices()` 函数列出了所有可用的物理 GPU 设备，并将其存储在变量 `gpus` 中。接下来，通过迭代 `gpus` 列表中的每个 GPU，使用 `set_memory_growth()` 函数将 GPU 内存设置为动态增长模式。这样可以使 TensorFlow 在需要时按需分配 GPU 内存，而不是一次性占用全部内存。