Get the list of visible physical devices.
See Migration guide for more details.
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
| ||(optional string) Only include devices matching this device type. For example "CPU" or "GPU".|
| List of visible |
© 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.