tf.config.list_logical_devices

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

tf.config.list_logical_devices(
    device_type=None
)

Используется в блокнотах

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

Вызов 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)