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.