TPU v2

本文档介绍了 Cloud TPU v2 的架构和支持的配置。

系统架构

如需了解 TPU v2 的架构详情和性能特征，请参阅用于训练深度神经网络的领域专用超级计算机。

配置

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

以下命令展示了如何创建具有 128 个 TensorCore 的 v2 TPU 切片：

  $ gcloud compute tpus tpu-vm create tpu-name \
    --zone=us-central1-a \
    --accelerator-type=v2-128 \
    --version=tpu-ubuntu2204-base

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

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