tf.config.experimental.list_logical_devices

Возвращает список логических устройств, созданных во время выполнения.

tf.config.experimental.list_logical_devices(
    device_type=None
)

Логические устройства могут соответствовать физическим устройствам или удалённым устройствам в кластере. Операции и тензоры могут быть размещены на этих устройствах с помощью name объекта LogicalDevice.

Например:

logical_devices = tf.config.experimental.list_logical_devices('GPU')
# 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)