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