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)