tf.config.experimental_connect_to_host

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

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

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

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

# Enable eager execution, and connect to the remote host.
tf.compat.v1.enable_eager_execution()
tf.contrib.eager.connect_to_remote_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)