Spec-Zone.ru › TensorFlow

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.

В следующем примере проверяется, что все видимые графические процессоры отключены:

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

Spec-Zone.ru

Настройки Оффлайн Что нового Помощь О нас
Spec-Zone .ru
спецификации, руководства, описания, API