tf.keras.distribution.list_devices
Return all the available devices based on the device type.
tf.keras.distribution.list_devices(
device_type=None
)
Note: in a distributed setting, global devices are returned.
| Args | |
|---|---|
device_type | string, one of "cpu", "gpu" or "tpu". Defaults to "gpu" or "tpu" if available when device_type is not provided. Otherwise will return the "cpu" devices. |
| Return | |
|---|---|
| List of devices that are available for distribute computation. |
© 2022 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 4.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/api_docs/python/tf/keras/distribution/list_devices
