Domande frequenti - Cloud TPU
Questo documento contiene un elenco di domande frequenti sulle Cloud TPU. it è suddiviso in sezioni:
- Domande frequenti indipendenti dal framework - domande sull'utilizzo delle Cloud TPU a prescindere dal framework ML in uso.
- Domande frequenti su JAX: domande sull'utilizzo di Cloud TPU con JAX.
- Domande frequenti su PyTorch: domande sull'utilizzo delle 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 che utilizza la TPU.
Come faccio ad aggiungere un volume del disco permanente a una VM Cloud TPU?
Per ulteriori informazioni, consulta Aggiungere un disco permanente a una VM TPU
Quali opzioni di archiviazione sono supportate/consigliate per l'addestramento con una VM TPU?
Per saperne di più, consulta le opzioni di archiviazione per Cloud TPU.
Domande frequenti su JAX
Come faccio a sapere se la TPU è utilizzata dal mio programma?
Esistono alcuni modi per verificare che JAX stia utilizzando la TPU:
Usa la funzione
jax.devices(). Ad esempio:assert jax.devices()[0].platform == 'tpu'Crea il profilo del programma e verifica che contenga operazioni TPU. Per scoprire di più, consulta la sezione Profilazione dei programmi JAX
Per ulteriori informazioni, leggi le domande frequenti su JAX.
Domande frequenti su Pytorch
Come faccio a sapere se la TPU è in uso dal mio programma?
Puoi eseguire i seguenti comandi Python:
>>> import torch_xla.core.xla_model as xm
>>> xm.get_xla_supported_devices(devkind="TPU")
Verifica se riesci a vedere dispositivi TPU.