tf.config.list_logical_devices

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

tf.config.list_logical_devices(
    device_type=None
)

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

Вызов tf.config.list_logical_devices заставляет runtime настроить любые tf.config.PhysicalDevice, видимые для runtime, тем самым предотвращая дальнейшую настройку. Чтобы избежать инициализации во время выполнения, используйте 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)