TPU v2

本文档介绍了 Cloud TPU v2。

系统架构

TPU v2 的架构详情和性能特性,请参阅 用于训练深度神经的领域特定超级计算机 广告网络

配置

TPU v2 Pod 由 512 个芯片组成,这些芯片通过可重新配置功能相互连接 或高速链接如需创建 TPU v2 Pod 切片,请在 TPU 创建命令 (gcloud compute tpus tpu-vm) 中使用 --accelerator-type 标志。您可以通过指定 TPU 版本和 TPU 核心数量来指定加速器类型。对于 例如,对于单个 v2 TPU,请使用 --accelerator-type=v2-8。对于 v2 Pod 切片 具有 128 个 TensorCore 时,请使用 --accelerator-type=v2-128

以下命令展示了如何创建包含 128 个 TensorCore 的 v2 TPU Pod 切片:

  $ gcloud compute tpus tpu-vm create tpu-name \
    --zone=zone \
    --accelerator-type=v2-128 \
    --version=tpu-vm-tf-2.17.0-pod-pjrt

如需详细了解如何管理 TPU,请参阅管理 TPU TPU。如需详细了解 TPU 系统, 请参阅系统 架构

下表列出了支持的 v2 TPU 类型:

TPU 版本 支持结束
v2-8 (结束日期尚未确定)
v2-32 (结束日期尚未确定)
v2-128 (结束日期尚未确定)
v2-256 (结束日期尚未确定)
v2-512 (结束日期尚未确定)