Crea un'istanza TPU

Questo documento mostra come creare un'istanza di macchina virtuale (VM) TPU, 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

  1. 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.

  2. Installa Google Cloud CLI.

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

  4. Per inizializzare gcloud CLI, esegui questo comando: 

    gcloud init

  5. 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.

    • 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 .

  6. 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.

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

  8. 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

  9. Installa Google Cloud CLI.

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

  11. Per inizializzare gcloud CLI, esegui questo comando: 

    gcloud init

  12. 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.

    • 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 .

  13. 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.

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

  15. 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
    16.

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:

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

Esegui questi comandi nel terminale locale o utilizzando Cloud Shell.

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

    gcloud compute instances create quickstart-tpu \
    --machine-type=ct6e-standard-4t \
    --zone=us-east5-b \
    --image-project=ubuntu-os-accelerator-images \
    --image-family=ubuntu-accelerator-2204-amd64-with-tpu-v6e

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

    gcloud compute instances describe quickstart-tpu --zone=us-east5-b

    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 --project=PROJECT_ID --zone=us-east5-b

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

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=us-east5-b

Passaggi successivi