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.17.0-pjrt
TPU の管理の詳細については、TPU の管理をご覧ください。TPU システム アーキテクチャの Cloud TPU の詳細については、システム アーキテクチャをご覧ください。
次の表に、サポートされている v2 TPU タイプを示します。
TPU バージョン | サポートの終了 |
---|---|
v2-8 | (終了日未定) |
v2-32 | (終了日未定) |
v2-128 | (終了日未定) |
v2-256 | (終了日未定) |
v2-512 | (終了日未定) |