TPU v2

本文件說明 Cloud TPU v2 的架構和支援的設定。

系統架構

如要瞭解 TPU v2 的架構詳細資料和效能特徵，請參閱「A Domain Specific Supercomputer for Training Deep Neural Networks」。

設定

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 類型：