tf.experimental.dtensor.initialize_tpu_system
Инициализация устройств TPU.
tf.experimental.dtensor.initialize_tpu_system(
enable_coordination_service=False
)
Эта функция выполняет дополнительную инициализацию, связанную с TPU, после вызова dtensor.initialize_multi_client для инициализации многоклиентского DTensor. Обратитесь к dtensor.initialize_multi_client для получения информации о соответствующих переменных среды, которые управляют инициализацией многоклиентского DTensor.
| Аргументы | |
|---|---|
enable_coordination_service | Если значение true, то включить службу распределённой координации, чтобы убедиться, что рабочие узлы знают устройства друг друга, что является необходимым условием для передачи данных через межрабочий обмен. |
| Исключения | |
|---|---|
RuntimeError | Если выполняется внутри tf.function. |
NotFoundError | Если в режиме eager не найдены устройства TPU. |
© 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/versions/r2.9/api_docs/python/tf/experimental/dtensor/initialize_tpu_system
