Spec-Zone.ru › TensorFlow

tf.config.LogicalDeviceConfiguration

Класс конфигурации для логических устройств.

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

Основные псевдонимы

tf.config.experimental.VirtualDeviceConfiguration

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

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

tf.compat.v1.config.LogicalDeviceConfiguration, tf.compat.v1.config.experimental.VirtualDeviceConfiguration

tf.config.LogicalDeviceConfiguration(
    memory_limit=None,
    experimental_priority=None,
    experimental_device_ordinal=None
)

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

Используется в руководстве Используется в учебных пособиях
  • Использование GPU
  • Генерация случайных чисел
  • Распределённое обучение с помощью Core API и DTensor
  • Основные понятия DTensor
  • Использование DTensor с Keras
  • Распределённое обучение с помощью DTensor
  • Распределённый ввод данных
  • Симуляции TFF с ускорителями
  • Блокнот с примечаниями к выпуску TFP (0.12.1)

Этот класс задаёт параметры конфигурации tf.config.PhysicalDevice при её инициализации в качестве tf.config.LogicalDevice во время инициализации среды выполнения. Не все поля допустимы для всех типов устройств.

Примеры использования см. в tf.config.get_logical_device_configuration и tf.config.set_logical_device_configuration.

Поля
memory_limit (необязательно) Максимальный объём памяти (в МБ), выделяемый виртуальному устройству. В настоящее время поддерживается только для GPU.
experimental_priority (необязательно) Приоритет, назначаемый виртуальному устройству. Чем меньше значение, тем выше приоритет; значение по умолчанию — 0. В рамках физического GPU планировщик GPU отдаёт предпочтение операциям на виртуальных устройствах с более высоким приоритетом. В настоящее время поддерживается только для GPU Nvidia.
experimental_device_ordinal (необязательно) Порядковый номер виртуального устройства. LogicalDevice с меньшим порядковым номером получит меньший идентификатор устройства. Для разрешения совпадений используются идентификатор физического устройства и его положение в списке. В настоящее время поддерживается только для GPU Nvidia.
Атрибуты
memory_limit Псевдоним namedtuple для поля с номером 0
experimental_priority Псевдоним namedtuple для поля с номером 1
experimental_device_ordinal Псевдоним namedtuple для поля с номером 2

© 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/LogicalDeviceConfiguration

Spec-Zone.ru

Настройки Оффлайн Что нового Помощь О нас
Spec-Zone .ru
спецификации, руководства, описания, API