tf.test.create_local_cluster

Создать и запустить локальные серверы и вернуть связанные Server объекты.

Просмотр псевдонимов

Альтернативные псевдонимы для миграции

См. Руководство по миграции для получения дополнительной информации.

tf.compat.v1.test.create_local_cluster

tf.test.create_local_cluster(
    num_workers: int,
    num_ps: int,
    protocol: str = 'grpc',
    worker_config: Optional[tf.compat.v1.ConfigProto] = None,
    ps_config: Optional[tf.compat.v1.ConfigProto] = None
) -> tuple[list[server_lib.Server], list[server_lib.Server]]

"PS" означает "параметр сервер": задача, ответственная за хранение и обновление параметров модели. Другие задачи отправляют обновления этих параметров, пока работают над оптимизацией параметров. Такое разделение труда между задачами не требуется, но часто используется для распределенного обучения.

Подробнее см. в https://www.tensorflow.org/guide/extend/architecture

components

Рисунок иллюстрирует взаимодействие этих компонентов. "/job:worker/task:0" и "/job:ps/task:0" — обе задачи со службами обработчика.

Пример:

workers, _ = tf.test.create_local_cluster(num_workers=2, num_ps=2)

worker_sessions = [tf.compat.v1.Session(w.target) for w in workers]

with tf.device("/job:ps/task:0"):
  ...
with tf.device("/job:ps/task:1"):
  ...
with tf.device("/job:worker/task:0"):
  ...
with tf.device("/job:worker/task:1"):
  ...

worker_sessions[0].run(...)

Аргументы
`num_workers`	Количество серверов обработчиков, которые необходимо запустить.
`num_ps`	Количество серверов PS, которые необходимо запустить.
`protocol`	Протокол связи. Допустимые значения описаны в документации по `tf.distribute.Server`.
`worker_config`	(необязательно) `tf.ConfigProto` для инициализации обработчиков. Может использоваться для создания нескольких устройств и т. д.
`ps_config`	(необязательно) `tf.ConfigProto` для инициализации серверов PS.

Возвращаемое значение
Кортеж `(worker_servers, ps_servers)`. `worker_servers` — список `num_workers` объектов типа `tf.distribute.Server` (все работают локально); и `ps_servers` — список `num_ps` объектов аналогичного типа.

Исключения
`ImportError`	если модуль portpicker не был найден во время загрузки

© 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/test/create_local_cluster