TPU v2

이 문서에서는 Cloud TPU v2의 아키텍처와 지원되는 구성을 설명합니다.

시스템 아키텍처

TPU v2의 아키텍처 세부정보 및 성능 특성은 심층신경망 학습용 도메인별 슈퍼 컴퓨터에서 확인할 수 있습니다.

구성

TPU v2 Pod는 재구성이 가능한 고속 링크로 상호 연결된 512개의 칩으로 구성되어 있습니다. TPU v2 포드 슬라이스를 만들려면 TPU 만들기 명령어(gcloud compute tpus tpu-vm)에 --accelerator-type 플래그를 사용합니다. TPU 버전과 TPU 코어 수를 지정하여 가속기 유형을 지정합니다. 예를 들어 단일 v2 TPU의 경우 --accelerator-type=v2-8을 사용합니다. TensorCore 128개가 포함된 v2 포드 슬라이스의 경우 --accelerator-type=v2-128을 사용합니다.

다음 명령어는 TensorCore 128개가 포함된 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 유형이 나열되어 있습니다.