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