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

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。非经特殊声明，原始代码版权归原作者所有，本译文未经允许或授权，请勿转载或复制。