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.18.0-pod-pjrt

TPU の管理の詳細については、TPU の管理をご覧ください。TPU システム アーキテクチャの Cloud TPU の詳細については、システム アーキテクチャをご覧ください。

次の表に、サポートされている v2 TPU タイプを示します。