W3cubDocs

/TensorFlow 2.4

tf.config.get_visible_devices

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

© 2020 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r2.4/api_docs/python/tf/config/get_visible_devices