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	（结束日期尚未确定）