Missed TensorFlow Dev Summit? Check out the video playlist. Watch recordings

tf.config.get_visible_devices

View source on GitHub

Get the list of visible physical devices.

tf.config.get_visible_devices(
    device_type=None
)

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') 
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 

Args:

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

Returns:

List of visible PhysicalDevices