Python 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 數量可能會發生變化。