tf.config.experimental_connect_to_host

Подключается к одному компьютеру для включения удалённого выполнения на нём.

tf.config.experimental_connect_to_host(
    remote_host=None, job_name='worker'
)

Сделает устройства на удалённом узле доступными для использования. Обратите внимание, что вызов этого более одного раза будет работать, но аннулирует все дескрипторы тензоров на старых удалённых устройствах.

Используя имя задачи по умолчанию worker, вы можете планировать операции для удалённого выполнения следующим образом:

# When eager execution is enabled, connect to the remote host.
tf.config.experimental_connect_to_host("exampleaddr.com:9876")

with ops.device("job:worker/replica:0/task:1/device:CPU:0"):
  # The following tensors should be resident on the remote device, and the op
  # will also execute remotely.
  x1 = array_ops.ones([2, 2])
  x2 = array_ops.ones([2, 2])
  y = math_ops.matmul(x1, x2)