TPU v2
本文档介绍了 Cloud TPU v2。
系统架构
TPU v2 的架构详情和性能特性,请参阅 用于训练深度神经的领域特定超级计算机 广告网络。
配置
TPU v2 Pod 由 512 个芯片组成,这些芯片通过可重新配置功能相互连接
或高速链接如需创建 TPU v2 Pod 切片,请使用 --accelerator-type
标志(在 TPU 创建命令 (gcloud compute tpus tpu-vm
) 中)。您可以指定
加速器类型。对于
例如,对于单个 v2 TPU,请使用 --accelerator-type=v2-8
。对于 v2 Pod 切片
具有 128 个 TensorCore 时,请使用 --accelerator-type=v2-128
。
以下命令展示了如何创建一个 128 的 v2 TPU Pod 切片 TensorCore:
$ gcloud compute tpus tpu-vm create tpu-name \
--zone=zone \
--accelerator-type=v2-128 \
--version=tpu-vm-tf-2.17.0-pjrt
如需详细了解如何管理 TPU,请参阅管理 TPU TPU。如需详细了解 TPU 系统, 请参阅系统 架构。
下表列出了支持的 v2 TPU 类型:
TPU 版本 | 支持结束 |
---|---|
v2-8 | (结束日期尚未确定) |
v2-32 | (结束日期尚未确定) |
v2-128 | (结束日期尚未确定) |
v2-256 | (结束日期尚未确定) |
v2-512 | (结束日期尚未确定) |