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.16.1-pjrt

TPU 관리에 대한 자세한 내용은 TPU 관리를 참조하세요. TPU 시스템 아키텍처 Cloud TPU에 대한 자세한 내용은 시스템 아키텍처를 참조하세요.

다음 표에는 지원되는 v2 TPU 유형이 나열되어 있습니다.

TPU 버전 지원 종료
v2-8 (종료일 미정)
v2-32 (종료일 미정)
v2-128 (종료일 미정)
v2-256 (종료일 미정)
v2-512 (종료일 미정)