当前位置: 首页>>代码示例 >>用法及示例精选 >>正文


Python tf.config.experimental_connect_to_host用法及代码示例


连接到单台机器以在其上启用远程执行。

用法

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

参数

  • remote_host host-port 格式的单个或列表远程服务器地址。
  • job_name 可以访问新服务器的作业名称。

抛出

  • ValueError 如果remote_host 为无。

将使远程主机上的设备可供使用。请注意,多次调用它会起作用,但会使旧远程设备上的任何张量句柄无效。

使用worker的默认job_name,您可以安排操作远程运行,如下所示:

# 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)

相关用法


注:本文由纯净天空筛选整理自tensorflow.org大神的英文原创作品 tf.config.experimental_connect_to_host。非经特殊声明,原始代码版权归原作者所有,本译文未经允许或授权,请勿转载或复制。