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
|
View source on GitHub