Crea un'istanza TPU

Crea un'istanza TPU

Questo documento mostra come creare un'istanza di macchina virtuale (VM) TPU v6e (Trillium), connettersi e eseguire un calcolo. L'istanza TPU che crei in questa guida rapida utilizza l'opzione di consumo on demand, quella predefinita.

Prima di iniziare

Accedi al tuo account Google Cloud . Se non conosci Google Cloud, crea un account per valutare le prestazioni dei nostri prodotti in scenari reali. I nuovi clienti ricevono anche 300 $di crediti senza costi per l'esecuzione, il test e il deployment dei workload.

Installa Google Cloud CLI.

Nota: se hai già installato gcloud CLI, assicurati di avere l'ultima versione eseguendo gcloud components update.

Se utilizzi un provider di identità (IdP) esterno, devi prima accedere a gcloud CLI con la tua identità federata.

Per inizializzare gcloud CLI, esegui questo comando:

gcloud init

Crea o seleziona un Google Cloud progetto.

Ruoli richiesti per selezionare o creare un progetto

• Seleziona un progetto: la selezione di un progetto non richiede un ruolo IAM specifico. Puoi selezionare qualsiasi progetto per il quale ti è stato concesso un ruolo.
• Crea un progetto: per creare un progetto, devi disporre del ruolo Autore progetto (roles/resourcemanager.projectCreator), che contiene l'autorizzazione resourcemanager.projects.create. Scopri come concedere i ruoli.

Nota: se non prevedi di conservare le risorse che crei in questa procedura, crea un progetto invece di selezionarne uno già esistente. Una volta completata questa procedura, puoi eliminare il progetto e tutte le relative risorse.

• Creare un progetto Google Cloud :

  gcloud projects create PROJECT_ID

  Sostituisci PROJECT_ID con un nome per il progetto Google Cloud che stai creando.

• Seleziona il progetto Google Cloud che hai creato:

  gcloud config set project PROJECT_ID

  Sostituisci PROJECT_ID con il nome del progetto Google Cloud .

Se utilizzi un progetto esistente per questa guida, verifica di disporre delle autorizzazioni necessarie per completare la guida. Se hai creato un nuovo progetto, disponi già delle autorizzazioni necessarie.

Verifica che la fatturazione sia attivata per il tuo progetto Google Cloud .

Abilita l'API Compute Engine:

Ruoli richiesti per abilitare le API

Per abilitare le API, devi disporre del ruolo IAM Amministratore utilizzo dei servizi (roles/serviceusage.serviceUsageAdmin), che include l'autorizzazione serviceusage.services.enable. Scopri come concedere i ruoli.

gcloud services enable compute.googleapis.com

Installa Google Cloud CLI.

Nota: se hai già installato gcloud CLI, assicurati di avere l'ultima versione eseguendo gcloud components update.

Se utilizzi un provider di identità (IdP) esterno, devi prima accedere a gcloud CLI con la tua identità federata.

Per inizializzare gcloud CLI, esegui questo comando:

gcloud init

Crea o seleziona un Google Cloud progetto.

Ruoli richiesti per selezionare o creare un progetto

• Seleziona un progetto: la selezione di un progetto non richiede un ruolo IAM specifico. Puoi selezionare qualsiasi progetto per il quale ti è stato concesso un ruolo.
• Crea un progetto: per creare un progetto, devi disporre del ruolo Autore progetto (roles/resourcemanager.projectCreator), che contiene l'autorizzazione resourcemanager.projects.create. Scopri come concedere i ruoli.

Nota: se non prevedi di conservare le risorse che crei in questa procedura, crea un progetto invece di selezionarne uno già esistente. Una volta completata questa procedura, puoi eliminare il progetto e tutte le relative risorse.

• Creare un progetto Google Cloud :

  gcloud projects create PROJECT_ID

  Sostituisci PROJECT_ID con un nome per il progetto Google Cloud che stai creando.

• Seleziona il progetto Google Cloud che hai creato:

  gcloud config set project PROJECT_ID

  Sostituisci PROJECT_ID con il nome del progetto Google Cloud .

Se utilizzi un progetto esistente per questa guida, verifica di disporre delle autorizzazioni necessarie per completare la guida. Se hai creato un nuovo progetto, disponi già delle autorizzazioni necessarie.

Verifica che la fatturazione sia attivata per il tuo progetto Google Cloud .

Abilita l'API Compute Engine:

Ruoli richiesti per abilitare le API

Per abilitare le API, devi disporre del ruolo IAM Amministratore utilizzo dei servizi (roles/serviceusage.serviceUsageAdmin), che include l'autorizzazione serviceusage.services.enable. Scopri come concedere i ruoli.

gcloud services enable compute.googleapis.com

Ruoli obbligatori

Per ottenere le autorizzazioni necessarie per creare un'istanza TPU, connetterti a essa utilizzando SSH ed eseguire comandi, chiedi all'amministratore di concederti i seguenti ruoli IAM sul progetto:

• Compute Instance Admin (v1) (roles/compute.instanceAdmin.v1)
• Utente Service Account (roles/iam.serviceAccountUser)
• Amministratore Service Usage (roles/serviceusage.serviceUsageAdmin)

Per saperne di più sulla concessione dei ruoli, consulta Gestisci l'accesso a progetti, cartelle e organizzazioni.

Potresti anche riuscire a ottenere le autorizzazioni richieste tramite i ruoli personalizzati o altri ruoli predefiniti.

Crea un'istanza TPU v6e (Trillium)

Esegui questi comandi nel terminale locale o utilizzando Cloud Shell.

1. Per creare un'istanza TPU v6e, esegui il seguente comando gcloud compute instances create:

   gcloud compute instances create quickstart-tpu \
       --machine-type=ct6e-standard-4t \
       --zone=europe-west4-a \
       --image-project=ubuntu-os-accelerator-images \
       --image-family=ubuntu-accel-2204-amd64-tpu-v5e-v5p-v6e \
       --maintenance-policy=TERMINATE
   

2. (Facoltativo) Verifica che l'istanza sia in esecuzione utilizzando il comando gcloud compute instances describe:

   gcloud compute instances describe quickstart-tpu --zone=europe-west4-a
   

   Nell'output, controlla il campo status per verificare se l'istanza è in esecuzione o ancora in attesa.

Connettiti all'istanza TPU

Puoi connetterti a un'istanza TPU utilizzando la console Google Cloud o gcloud CLI.

Per connetterti alla VM tramite SSH, esegui il seguente comando gcloud compute ssh:

gcloud compute ssh quickstart-tpu --zone=europe-west4-a

Installare JAX

Installa il framework JAX eseguendo questo comando sull'istanza TPU:

pip install jax[tpu] -f https://storage.googleapis.com/jax-releases/libtpu_releases.html

Nota: se questo comando non va a buon fine, potrebbe essere necessario eseguire sudo apt update e sudo apt install python3-pip.

Eseguire un controllo del sistema

Verifica che JAX possa accedere alla TPU ed eseguire operazioni di base eseguendo i seguenti comandi sull'istanza TPU:

1. Avvia l'interprete Python 3:

   python3
   

2. Importa il modulo JAX:

   >>> import jax
   

3. Visualizza il numero di core TPU disponibili:

   >>> jax.device_count()
   

   L'output mostra il numero di core TPU. Il numero di core dipende dalla versione della TPU che stai utilizzando. Per saperne di più, consulta Versioni TPU.

Eseguire un calcolo

1. Esegui questo comando per sommare due numeri:

   >>> jax.numpy.add(1, 1)
   

   L'output dovrebbe essere simile al seguente:

   >>> Array(2, dtype=int32, weak_type=True)
   

2. Esci dall'interprete Python:

   >>> 
   exit()
   

Esegui la pulizia

Per evitare che al tuo account Google Cloud vengano addebitati costi relativi alle risorse utilizzate in questa pagina, elimina il progetto Google Cloud con le risorse.

Per evitare che al tuo account Google Cloud vengano addebitati costi relativi alle risorse utilizzate in questa pagina, elimina l'istanza TPU utilizzando il comando gcloud compute instances delete:

gcloud compute instances delete quickstart-tpu --zone=europe-west4-a

Passaggi successivi

• Scopri di più sulle risorse Cloud TPU in Compute Engine
• Crea un'istanza Cloud TPU utilizzando altre opzioni di consumo
• Crea istanze Cloud TPU con i gruppi di istanze gestite