tf.config.get_visible_devices

Получить список видимых физических устройств.

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

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

tf.config.experimental.get_visible_devices

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

См. Руководство по миграции для получения дополнительных сведений.

tf.compat.v1.config.experimental.get_visible_devices, tf.compat.v1.config.get_visible_devices

tf.config.get_visible_devices(
    device_type=None
)

Возвращает список PhysicalDevice, которые в данный момент помечены как видимые для среды выполнения. У видимого устройства будет по крайней мере один LogicalDevice, связанный с ним после инициализации среды выполнения.

Следующий пример проверяет, что все видимые GPU были отключены:

physical_devices = tf.config.list_physical_devices('GPU')
try:
  # Disable all GPUS
  tf.config.set_visible_devices([], 'GPU')
  visible_devices = tf.config.get_visible_devices()
  for device in visible_devices:
    assert device.device_type != 'GPU'
except:
  # Invalid device or cannot modify virtual devices once initialized.
  pass

Аргументы
`device_type`	(необязательная строка) Указывает только устройства, соответствующие этому типу устройства. Например, "CPU" или "GPU".

Возвращаемое значение
Список видимых `PhysicalDevice`

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