tf.config.experimental.get_memory_growth

Получить, если для физического устройства включен рост памяти.

tf.config.experimental.get_memory_growth(
    device
)

Физическое устройство с настройкой роста памяти не будет выделять всю память на устройстве сразу.

Например:

physical_devices = config.experimental.list_physical_devices('GPU')
assert len(physical_devices) > 0, "Not enough GPU hardware devices available"
tf.config.experimental.set_memory_growth(physical_devices[0], True)
assert tf.config.experimental.get_memory_growth(physical_devices[0]) == True