Pertanyaan Umum (FAQ) - Cloud TPU
Dokumen ini berisi daftar pertanyaan umum (FAQ) tentang Cloud TPU. Laporan ini dibagi menjadi beberapa bagian:
- FAQ independen framework - pertanyaan tentang penggunaan Cloud TPU, terlepas dari framework ML yang Anda gunakan.
- FAQ JAX - pertanyaan tentang penggunaan Cloud TPU dengan JAX.
- FAQ PyTorch - pertanyaan tentang penggunaan Cloud TPU dengan PyTorch.
FAQ independen framework
Bagaimana cara memeriksa proses yang menggunakan TPU di VM Cloud TPU?
Jalankan sudo lsof -w /dev/accel*
di VM Cloud TPU untuk mencetak ID proses dan
informasi lain tentang proses yang menggunakan TPU.
Bagaimana cara menambahkan volume persistent disk ke VM Cloud TPU?
Untuk mengetahui informasi selengkapnya, lihat Menambahkan persistent disk ke VM TPU
Opsi penyimpanan apa yang didukung/direkomendasikan untuk pelatihan dengan VM TPU?
Untuk mengetahui informasi selengkapnya, lihat Opsi penyimpanan Cloud TPU.
FAQ JAX
Bagaimana cara mengetahui apakah TPU sedang digunakan oleh program saya?
Ada beberapa cara untuk memeriksa kembali apakah JAX menggunakan TPU:
Gunakan fungsi
jax.devices()
. Contoh:assert jax.devices()[0].platform == 'tpu'
Buat profil program Anda dan verifikasi bahwa profil tersebut berisi operasi TPU. Untuk informasi selengkapnya, lihat Membuat profil program JAX
Untuk informasi selengkapnya, lihat FAQ JAX
FAQ Pytorch
Bagaimana cara mengetahui apakah TPU sedang digunakan oleh program saya?
Anda dapat menjalankan perintah python berikut:
>>> import torch_xla.core.xla_model as xm
>>> xm.get_xla_supported_devices(devkind="TPU")
Dan verifikasi apakah Anda dapat melihat perangkat TPU.