tf.config.experimental.set_memory_growth

Указывает, следует ли включить рост использования памяти для PhysicalDevice.

Просмотреть псевдонимы

Совместимые псевдонимы для миграции

Подробнее см. в руководстве по миграции.

tf.compat.v1.config.experimental.set_memory_growth

tf.config.experimental.set_memory_growth(
    device, enable
)

Используется в блокнотах

Используется в руководстве	Используется в учебных материалах
Использование GPU Импорт модели JAX с помощью JAX2TF Использование формата SavedModel	Демонстрация Universal Sentence Encoder SentEval Байесовская модель гауссовой смеси и гамильтонова MCMC Моделирование распространения COVID-19 в Европе и влияния мер вмешательства

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

Например:

physical_devices = tf.config.list_physical_devices('GPU')
try:
  tf.config.experimental.set_memory_growth(physical_devices[0], True)
except:
  # Invalid device or cannot modify virtual devices once initialized.
  pass

Аргументы
`device`	`PhysicalDevice` для настройки
`enable`	(логическое значение) Включить или отключить рост использования памяти

Вызывает исключения
`ValueError`	Указан недопустимый `PhysicalDevice`.
`RuntimeError`	Среда выполнения уже инициализирована.

© 2022 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 4.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/api_docs/python/tf/config/experimental/set_memory_growth