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