Функция config.experimental.get_memory_growth

Функция config.experimental.get_memory_growth позволяет узнать, включен ли режим динамического роста памяти для конкретного физического устройства. Первым параметром функция принимает объект устройства, полученный через config.list_physical_devices. Возвращает логическое значение: True, если рост памяти включен, и False в противном случае.

Синтаксис

tf.config.experimental.get_memory_growth(device)

Пример

Давайте проверим, включен ли режим динамического выделения памяти для первого доступного GPU:

import tensorflow as tf

gpus = tf.config.list_physical_devices('GPU')

if gpus:
	res = tf.config.experimental.get_memory_growth(gpus[0])
	print(res)
else:
	print("No GPU available")

Результат выполнения кода:

False

Пример

Давайте включим динамический рост памяти и проверим его состояние:

import tensorflow as tf

gpus = tf.config.list_physical_devices('GPU')

if gpus:
	tf.config.experimental.set_memory_growth(gpus[0], True)
	res = tf.config.experimental.get_memory_growth(gpus[0])
	print(res)
else:
	print("No GPU available")

Результат выполнения кода:

True

Смотрите также

функцию experimental.set_memory_growth,
которая включает динамическое выделение памяти
функцию config.list_physical_devices,
которая возвращает список физических устройств
функцию config.list_logical_devices,
которая возвращает список логических устройств
функцию config.get_visible_devices,
которая возвращает список видимых устройств