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, доступных среде выполнения, может измениться во время запуска среды выполнения из-за того, что некоторые устройства помечены как невидимые или настраиваются несколько логических устройств.