RSVP for your your local TensorFlow Everywhere event today!


View source on GitHub

Get the list of visible physical devices.

Returns the list of PhysicalDevices currently marked as visible to the runtime. A visible device will have at least one LogicalDevice associated with it once the runtime is initialized.

The following example verifies all visible GPUs have been disabled:

physical_devices = tf.config.list_physical_devices('GPU')
  # 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'
  # Invalid device or cannot modify virtual devices once initialized.

device_type (optional string) Only include devices matching this device type. For example "CPU" or "GPU".

List of visible PhysicalDevices