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

Python tf.distribute.cluster_resolver.TPUClusterResolver.connect用法及代碼示例


用法

@staticmethod
connect(
    tpu=None, zone=None, project=None
)

參數

  • tpu 與要使用的 TPU 對應的字符串。它可以是 TPU 名稱或 TPU worker gRPC 地址。如果未設置，它將嘗試自動解析 Cloud TPU 上的 TPU 地址。
  • zone TPU 所在的區域。如果省略或為空，我們將假設 TPU 的區域與 GCE 虛擬機的區域相同，我們將嘗試從 GCE 元數據服務中發現。
  • project 包含 Cloud TPU 的 GCP 項目的名稱。如果省略或為空，我們將嘗試從 GCE 元數據服務中發現 GCE VM 的項目名稱。

返回

  • TPUClusterResolver 對象的一個 實例。

拋出

  • NotFoundError 如果在 Eager 模式下找不到 TPU 設備。

初始化 TPU 並返回一個 TPUClusterResolver。

此 API 將連接到遠程 TPU 集群並初始化 TPU 硬件。示例用法：

resolver = tf.distribute.cluster_resolver.TPUClusterResolver.connect(
    tpu='')

它可以被視為以下代碼的方便包裝器：

resolver = tf.distribute.cluster_resolver.TPUClusterResolver(tpu='')
tf.config.experimental_connect_to_cluster(resolver)
tf.tpu.experimental.initialize_tpu_system(resolver)

相關用法


注：本文由純淨天空篩選整理自tensorflow.org大神的英文原創作品 tf.distribute.cluster_resolver.TPUClusterResolver.connect。非經特殊聲明，原始代碼版權歸原作者所有，本譯文未經允許或授權，請勿轉載或複製。