tf.config.get_visible_devices
Stay organized with collections
Save and categorize content based on your preferences.
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
|
Except as otherwise noted, the content of this page is licensed under the Creative Commons Attribution 4.0 License, and code samples are licensed under the Apache 2.0 License. For details, see the Google Developers Site Policies. Java is a registered trademark of Oracle and/or its affiliates. Some content is licensed under the numpy license.
Last updated 2023-10-06 UTC.
[null,null,["Last updated 2023-10-06 UTC."],[],[],null,["# tf.config.get_visible_devices\n\n\u003cbr /\u003e\n\n|--------------------------------------------------------------------------------------------------------------------------------|\n| [View source on GitHub](https://github.com/tensorflow/tensorflow/blob/v2.14.0/tensorflow/python/framework/config.py#L494-L524) |\n\nGet the list of visible physical devices.\n\n#### View aliases\n\n\n**Main aliases**\n\n[`tf.config.experimental.get_visible_devices`](https://www.tensorflow.org/api_docs/python/tf/config/get_visible_devices)\n**Compat aliases for migration**\n\nSee\n[Migration guide](https://www.tensorflow.org/guide/migrate) for\nmore details.\n\n[`tf.compat.v1.config.experimental.get_visible_devices`](https://www.tensorflow.org/api_docs/python/tf/config/get_visible_devices), [`tf.compat.v1.config.get_visible_devices`](https://www.tensorflow.org/api_docs/python/tf/config/get_visible_devices)\n\n\u003cbr /\u003e\n\n tf.config.get_visible_devices(\n device_type=None\n )\n\nReturns the list of `PhysicalDevice`s currently marked as visible to the\nruntime. A visible device will have at least one `LogicalDevice` associated\nwith it once the runtime is initialized.\n\nThe following example verifies all visible GPUs have been disabled: \n\n physical_devices = tf.config.list_physical_devices('GPU')\n try:\n # Disable all GPUS\n tf.config.set_visible_devices([], 'GPU')\n visible_devices = tf.config.get_visible_devices()\n for device in visible_devices:\n assert device.device_type != 'GPU'\n except:\n # Invalid device or cannot modify virtual devices once initialized.\n pass\n\n\u003cbr /\u003e\n\n\u003cbr /\u003e\n\n\u003cbr /\u003e\n\n| Args ---- ||\n|---------------|-----------------------------------------------------------------------------------------------|\n| `device_type` | (optional string) Only include devices matching this device type. For example \"CPU\" or \"GPU\". |\n\n\u003cbr /\u003e\n\n\u003cbr /\u003e\n\n\u003cbr /\u003e\n\n\u003cbr /\u003e\n\n| Returns ------- ||\n|---|---|\n| List of visible `PhysicalDevice`s ||\n\n\u003cbr /\u003e"]]
View source on GitHub