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 Распределенное обучение с базовыми API и DTensor Концепции DTensor Генерация случайных чисел	Использование DTensors с Keras Распределенное обучение с DTensors Распределенный ввод Симуляции TFF с ускорителями Заметки о выпуске TFP (ноутбук 0.12.1)

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

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

Следующий пример разбивает ЦП на 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

Следующий пример разбивает графический процессор на 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