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. Installa Google Cloud CLI.

  2. Configura gcloud CLI per utilizzare la tua identità federata.

    Per ulteriori informazioni, vedi Accedi a gcloud CLI con la tua identità federata.

  3. Per inizializzare gcloud CLI, esegui questo comando: 

    gcloud init

  4. Crea o seleziona un Cloud de Confiance 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 Cloud de Confiance :

      gcloud projects create PROJECT_ID

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

    • Seleziona il progetto Cloud de Confiance che hai creato:

      gcloud config set project PROJECT_ID

      Sostituisci PROJECT_ID con il nome del progetto Cloud de Confiance .

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

  6. Verifica che la fatturazione sia attivata per il tuo progetto Cloud de Confiance .

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

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 Cloud de Confiance 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 Cloud de Confiance vengano addebitati costi relativi alle risorse utilizzate in questa pagina, elimina il progetto Cloud de Confiance con le risorse.

Per evitare che al tuo account Cloud de Confiance by S3NS 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