tf.config.list_physical_devices
Возвращает список физических устройств, доступных среде выполнения на хосте.
tf.config.list_physical_devices(
device_type=None
)
Используется в блокнотах
| Используется в руководстве | Используется в учебных материалах |
|---|---|
Физические устройства — это аппаратные устройства, имеющиеся на хост-компьютере. По умолчанию все обнаруженные устройства CPU и GPU считаются доступными.
Этот API позволяет запрашивать физические аппаратные ресурсы до инициализации среды выполнения. Это даёт возможность вызывать дополнительные API настройки. В отличие от tf.config.list_logical_devices, этот вызов инициирует среду выполнения, чтобы вывести список настроенных устройств.
В следующем примере выводится количество доступных GPU на хосте.
physical_devices = tf.config.list_physical_devices('GPU')
print("Num GPUs:", len(physical_devices))
Num GPUs: ...Однако количество GPU, доступных среде выполнения, может измениться во время её инициализации, если некоторые устройства помечаются как недоступные или настраиваются несколько логических устройств.
| Аргументы | |
|---|---|
device_type | (необязательная строка) Включать только устройства, соответствующие этому типу устройства. Например, «CPU» или «GPU». |
Notes |
|
| Возвращает | |
|---|---|
Список обнаруженных объектов tf.config.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/list_physical_devices