將 TPU 連線至共用虛擬私有雲網路
設定虛擬私有雲主專案
您必須授予服務專案中的 TPU 服務帳戶權限,才能管理主機專案中的資源。您可以使用「TPU 共用虛擬私有雲代理程式」(roles/tpu.xpnAgent
) 角色執行這項操作。執行下列 gcloud
指令,即可授予此角色繫結。
gcloud projects add-iam-policy-binding host-project-id \ --member=serviceAccount:service-your-service-project-number@gcp-sa-tpu.iam.gserviceaccount.com \ --role=roles/tpu.xpnAgent
建立與共用虛擬私有雲網路連線的 TPU VM
首先,請確定區域中可用的加速器類型和版本
gcloud compute tpus accelerator-types list --zone zone
gcloud compute tpus versions list --zone zone
建立 TPU 時,您會將 TPU VM 連線至共用虛擬私有雲網路。使用 --network
標記指定共用 VPC:
gcloud compute tpus tpu-vm create tpu-name \ --zone zone \ --accelerator-type accelerator-type \ --network projects/host-project-id/global/networks/host-network \ --version tpu-image-version \ --project your-service-project-id
您可以使用 gcloud describe
指令,確認 TPU VM 是否已連線至共用 VPC:
$ gcloud compute tpus tpu-vm describe tpu-name --zone zone
回應中會包含 TPU VM 連結的網路:
acceleratorType: v3-8 apiVersion: V2 cidrBlock: 10.128.0.0/20 createTime: '2022-06-17T21:32:13.859274143Z' health: HEALTHY id: '0000000000000000000' name: projects/my-project/locations/us-central1-b/nodes/my-tpu networkConfig: enableExternalIps: true network: projects/my-project/global/networks/default subnetwork: projects/my-project/regions/us-central1/subnetworks/default networkEndpoints: - accessConfig: externalIp: 000.000.000.000 ipAddress: 10.128.0.104 port: 8470 runtimeVersion: tpu-vm-tf-2.8.0 schedulingConfig: {} serviceAccount: email: 00000000000-compute@developer.gserviceaccount.com scope: - https://www.googleapis.com/auth/devstorage.read_write - https://www.googleapis.com/auth/logging.write - https://www.googleapis.com/auth/service.management - https://www.googleapis.com/auth/servicecontrol - https://www.googleapis.com/auth/cloud-platform - https://www.googleapis.com/auth/pubsub shieldedInstanceConfig: {} state: READY
刪除 TPU VM
使用完 TPU VM 後,請務必刪除。
gcloud compute tpus tpu-vm delete tpu-name --zone zone