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