Mantieni tutto organizzato con le raccolte Salva e classifica i contenuti in base alle tue preferenze.

Domande frequenti - Cloud TPU

Questo documento contiene un elenco di domande frequenti sulle Cloud TPU. suddivise in sezioni:

  1. Domande frequenti indipendenti dal framework: domande sull'utilizzo delle Cloud TPU, indipendentemente dal framework ML utilizzato.
  2. Domande frequenti su JAX: domande sull'utilizzo delle Cloud TPU con JAX.
  3. Domande frequenti su PyTorch: domande sull'utilizzo di Cloud TPU con PyTorch.

Domande frequenti indipendenti dal framework

Come posso controllare quale processo utilizza la TPU su una VM Cloud TPU?

Esegui sudo lsof -w /dev/accel* sulla VM Cloud TPU per stampare l'ID processo e altre informazioni sul processo utilizzando la TPU.

Come faccio ad aggiungere un volume del disco permanente a una VM Cloud TPU?

Per saperne di più, vedi Aggiungere un disco permanente a una VM TPU

Quali opzioni di archiviazione sono supportate/consigliate per l'addestramento con VM TPU?

Per saperne di più, vedi Opzioni di archiviazione di Cloud TPU.

Domande frequenti su JAX

Come posso sapere se la TPU è utilizzata dal mio programma?

Ci sono vari modi per verificare che JAX utilizzi la TPU:

  1. Usa la funzione jax.devices(). Ad esempio:

    assert jax.devices()[0].platform == 'tpu'
    
  2. Profila il tuo programma e verifica che il profilo contenga le operazioni TPU. Per ulteriori informazioni, consulta la sezione Profilazione dei programmi JAX

Per ulteriori informazioni, consulta le Domande frequenti su JAX

Domande frequenti su Pytorch

Come posso sapere se la TPU è utilizzata dal mio programma?

Puoi eseguire i seguenti comandi Python:

>>> import torch_xla.core.xla_model as xm
>>> xm.get_xla_supported_devices(devkind="TPU")

e verifica se riesci a vedere i dispositivi TPU.