tf.experimental.dtensor.initialize_multi_client

Инициализирует многоклиентский DTensor.

tf.experimental.dtensor.initialize_multi_client(
    enable_coordination_service: Optional[bool] = False
) -> None

Следующие переменные среды управляют поведением этой функции. Если переменные не заданы, DTensor будет настроен на работу в одноклиентском режиме.

• DTENSOR_CLIENT_ID: целое число, от 0 до num_clients - 1, для идентификации идентификатора клиента текущего процесса.
• DTENSOR_NUM_CLIENTS: целое число, количество клиентов.
• DTENSOR_JOB_NAME: строка, строка типа имени хоста для имени задачи dtensor. Имя задачи используется TensorFlow в разделе имени задачи DeviceSpec.
• DTENSOR_JOBS: строка, список, разделенный запятыми. Каждый элемент списка имеет формат {hostname}:{port} и элементы должны быть отсортированы в алфавитном порядке. Это означает, что номера портов RPC клиентов с одного хоста должны быть упорядочены по идентификатору клиента. Примеры допустимых значений DTENSOR_JOBS:
  • 4 клиента на localhost: localhost:10000,localhost:10001,localhost:10002,localhost:10003
  • 2 клиента на host1, 2 клиента на host2 host1:10000,host1:10001,host2:10000,host2:10003