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
