tf.config.set_visible_devices

Задать список видимых устройств.

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

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

tf.config.experimental.set_visible_devices

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

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

tf.compat.v1.config.experimental.set_visible_devices, tf.compat.v1.config.set_visible_devices

tf.config.set_visible_devices(
    devices, device_type=None
)

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

Используется в руководстве
Использование GPU

Указывает, какие объекты PhysicalDevice видимы среде выполнения. TensorFlow будет выделять память и размещать операции только на видимых физических устройствах, поскольку в противном случае на них не будет создано ни одного LogicalDevice. По умолчанию все обнаруженные устройства считаются видимыми.

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

physical_devices = tf.config.list_physical_devices('GPU')
try:
  # Disable first GPU
  tf.config.set_visible_devices(physical_devices[1:], 'GPU')
  logical_devices = tf.config.list_logical_devices('GPU')
  # Logical device was not created for first GPU
  assert len(logical_devices) == len(physical_devices) - 1
except:
  # Invalid device or cannot modify virtual devices once initialized.
  pass

Аргументы
`devices`	Список устройств `PhysicalDevice`, которые следует сделать видимыми
`device_type`	(необязательно) Настраивать только устройства, соответствующие этому типу устройства. Например, «CPU» или «GPU». Остальные устройства останутся без изменений.

Вызывает исключение
`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_visible_devices