tf.config.set_logical_device_configuration

Задаёт конфигурацию логического устройства для tf.config.PhysicalDevice.

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

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

tf.config.experimental.set_virtual_device_configuration

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

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

tf.compat.v1.config.experimental.set_virtual_device_configuration, tf.compat.v1.config.set_logical_device_configuration

tf.config.set_logical_device_configuration(
    device, logical_devices
)

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

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

По умолчанию видимое tf.config.PhysicalDevice будет связано с одним tf.config.LogicalDevice после инициализации среды выполнения. Указание списка объектов tf.config.LogicalDeviceConfiguration позволяет создать несколько устройств на одном tf.config.PhysicalDevice.

Конфигурации логических устройств можно изменять вызовом этой функции, пока среда выполнения не инициализирована. Если вызвать эту функцию после инициализации среды выполнения, будет возникнуто исключение RuntimeError.

В следующем примере CPU разделяется на 2 логических устройства:

physical_devices = tf.config.list_physical_devices('CPU')
assert len(physical_devices) == 1, "No CPUs found"
# Specify 2 virtual CPUs. Note currently memory limit is not supported.
try:
  tf.config.set_logical_device_configuration(
    physical_devices[0],
    [tf.config.LogicalDeviceConfiguration(),
     tf.config.LogicalDeviceConfiguration()])
  logical_devices = tf.config.list_logical_devices('CPU')
  assert len(logical_devices) == 2

  tf.config.set_logical_device_configuration(
    physical_devices[0],
    [tf.config.LogicalDeviceConfiguration(),
     tf.config.LogicalDeviceConfiguration(),
     tf.config.LogicalDeviceConfiguration(),
     tf.config.LogicalDeviceConfiguration()])
except:
  # Cannot modify logical devices once initialized.
  pass

В следующем примере GPU разделяется на 2 логических устройства по 100 МБ каждое:

physical_devices = tf.config.list_physical_devices('GPU')
try:
  tf.config.set_logical_device_configuration(
    physical_devices[0],
    [tf.config.LogicalDeviceConfiguration(memory_limit=100),
     tf.config.LogicalDeviceConfiguration(memory_limit=100)])

  logical_devices = tf.config.list_logical_devices('GPU')
  assert len(logical_devices) == len(physical_devices) + 1

  tf.config.set_logical_device_configuration(
    physical_devices[0],
    [tf.config.LogicalDeviceConfiguration(memory_limit=10),
     tf.config.LogicalDeviceConfiguration(memory_limit=10)])
except:
  # Invalid device or cannot modify logical devices once initialized.
  pass

Аргументы
`device`	`PhysicalDevice` для настройки.
`logical_devices`	(необязательно) Список объектов `tf.config.LogicalDeviceConfiguration`, которые следует выделить для указанного `PhysicalDevice`. Если значение равно None, будет использована конфигурация по умолчанию.

Исключения
`ValueError`	Если проверка аргументов завершилась неудачно.
`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/set_logical_device_configuration