Crea un'istanza TPU

Crea un'istanza TPU

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

Prima di iniziare

Accedi al tuo Google Cloud account. Se non hai mai utilizzato 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 carichi di lavoro.

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 su cui ti è stato concesso un ruolo.
• Crea un progetto: per creare un progetto, devi disporre del ruolo Autore progetto (roles/resourcemanager.projectCreator), che contiene l' resourcemanager.projects.create autorizzazione. 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.

• Crea un Google Cloud progetto:

  gcloud projects create PROJECT_ID

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

• Seleziona il Google Cloud progetto che hai creato:

  gcloud config set project PROJECT_ID

  Sostituisci PROJECT_ID con il nome del Google Cloud progetto.

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 Google Cloud progetto.

Abilita l'API Compute Engine:

Ruoli richiesti per abilitare le API

Per abilitare le API, devi disporre del ruolo IAM Amministratore Service Usage (roles/serviceusage.serviceUsageAdmin), che contiene l' serviceusage.services.enable autorizzazione. 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 su cui ti è stato concesso un ruolo.
• Crea un progetto: per creare un progetto, devi disporre del ruolo Autore progetto (roles/resourcemanager.projectCreator), che contiene l' resourcemanager.projects.create autorizzazione. 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.

• Crea un Google Cloud progetto:

  gcloud projects create PROJECT_ID

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

• Seleziona il Google Cloud progetto che hai creato:

  gcloud config set project PROJECT_ID

  Sostituisci PROJECT_ID con il nome del Google Cloud progetto.

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 Google Cloud progetto.

Abilita l'API Compute Engine:

Ruoli richiesti per abilitare le API

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

gcloud services enable compute.googleapis.com

Ruoli obbligatori

Per ottenere le autorizzazioni necessarie per creare un'istanza TPU, connetterti utilizzando SSH ed eseguire comandi, chiedi all'amministratore di concederti i seguenti ruoli IAM sul tuo 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 i seguenti comandi nel terminale locale o utilizzando Cloud Shell.

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

   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 gcloud compute instances describe comando:

   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 Google Cloud console o la gcloud CLI.

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

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

Installa JAX

Installa il framework JAX eseguendo il seguente 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.

Esegui 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 maggiori informazioni, consulta la sezione Versioni TPU.

Esegui un calcolo

1. Esegui il seguente 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()
   

Libera spazio

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

In alternativa, se vuoi conservare il progetto, puoi eliminare solo la VM 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 MIG