tf.config.get_visible_devices

Получить список видимых физических устройств.

tf.config.get_visible_devices(
    device_type=None
)

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

Следующий пример проверяет, были ли отключены все видимые GPU:

physical_devices = tf.config.list_physical_devices('GPU')
try:
  # Disable all GPUS
  tf.config.set_visible_devices([], 'GPU')
  visible_devices = tf.config.get_visible_devices()
  for device in visible_devices:
    assert device.device_type != 'GPU'
except:
  # Invalid device or cannot modify virtual devices once initialized.
  pass