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