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。对于具有 128 个 TensorCore 的 v2 Pod 切片，请使用 --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.16.1-pjrt

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

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

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