當前位置: 首頁>>代碼示例 >>用法及示例精選 >>正文


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。非經特殊聲明,原始代碼版權歸原作者所有,本譯文未經允許或授權,請勿轉載或複製。