 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 GPUStf.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 |