tf.keras.distribution.list_devices
Возвращает все доступные устройства на основе типа устройства.
tf.keras.distribution.list_devices(
device_type=None
)
Примечание: в распределённой среде возвращаются глобальные устройства.
| Аргументы | |
|---|---|
device_type | строка, одна из "cpu", "gpu" или "tpu". По умолчанию "gpu" или "tpu", если доступно, когда device_type не предоставлено. В противном случае возвращаются устройства "cpu". |
| Возвращаемое значение | |
|---|---|
| Список устройств, доступных для распределённых вычислений. |
© 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
