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 数量可能会发生变化。