Python tf.config.list_logical_devices用法及代码示例

返回运行时创建的逻辑设备列表。

用法

tf.config.list_logical_devices(
    device_type=None
)

逻辑设备可以对应集群中的物理设备或远程设备。可以使用 tf.config.LogicalDevice 的 name 将操作和张量放置在这些设备上。

调用tf.config.list_logical_devices 会触发运行时配置运行时可见的任何tf.config.PhysicalDevice，从而阻止进一步的配置。为避免运行时初始化，请改为调用tf.config.list_physical_devices。

例如：

logical_devices = tf.config.list_logical_devices('GPU')
if len(logical_devices) > 0:
  # Allocate on GPU:0
  with tf.device(logical_devices[0].name):
    one = tf.constant(1)
  # Allocate on GPU:1
  with tf.device(logical_devices[1].name):
    two = tf.constant(2)