tf.config.experimental.get_memory_info

Получает информацию о памяти выбранного устройства в виде словаря.

tf.config.experimental.get_memory_info(
    device
)

Эта функция возвращает словарь, содержащий информацию об использовании памяти устройства. Например:

if tf.config.list_physical_devices('GPU'):
  # Returns a dict in the form {'current': <current mem usage>,
  #                             'peak': <peak mem usage>}
  tf.config.experimental.get_memory_info('GPU:0')

В настоящее время возвращаются следующие ключи:

• 'current': текущий объем памяти, используемый устройством, в байтах.
• 'peak': пиковый объем памяти, использованный устройством за время выполнения программы, в байтах. Это значение можно сбросить с помощью tf.config.experimental.reset_memory_stats.

В будущем могут быть добавлены другие ключи, в том числе зависящие от устройства.

В настоящее время поддерживаются только GPU и TPU. При вызове для устройства CPU возникает исключение.

Для GPU TensorFlow по умолчанию выделяет всю память, если это не изменено с помощью tf.config.experimental.set_memory_growth. Словарь содержит только текущий и пиковый объем памяти, фактически используемый TensorFlow, а не объем памяти, выделенный TensorFlow на GPU.