Pytorch Lightning'i TPU'larla kullanmak için birkaç özel bağımlılık ve konfigürasyon gereklidir:
1. Pytorch XLA: Bu kütüphane, Pytorch'u TPU'larla arayüz etmek için çok önemlidir. Tipik olarak `! PIP yüklemesi Cloud-tpu-istemcisi == 0.10` gibi komutlar kullanılarak yapılan Pytorch XLA tekerleğini kurmanız gerekir ve ardından Google'ın depolamasından uygun Pytorch XLA tekerleğini `` Torch_XLA-1.9-CP37- CP37M-LINUX_X86_64.WHL` [3] [5].
2. Bulut TPU istemcisi: Google Colab [3] [7] gibi ortamlarda TPU'larla etkileşim kurmak için `Cloud-tpu-istemci == 0.10 'yükleme gereklidir.
3. Pytorch ve Torchvision: Pytorch ve Torchvision'un TPU ortamı ile kurulduğundan ve uyumlu olduğundan emin olun. Bu genellikle TPU ortamını oluşturduktan sonra bu paketlerin yeniden yüklenmesini içerir [1] [7].
4. Sürümün kurulumunuzla uyumlu olduğundan emin olun [1] [7].
5. XRT TPU Cihaz Yapılandırması: Bulut TPU VM'leri için XRT TPU aygıt yapılandırmasını `` XRT_TPU_CONFIG = "LocalService Dışa Aktarma; 0; LocalHost: 51011" `[5] gibi komutları kullanarak ayarlamanız gerekir.
6. Dağıtılmış Eğitim: Pytorch Lightning, TPU'larda dağıtılmış eğitimi otomatik olarak işler, bu nedenle dağıtılmış örnekleyiciler için ek yapılandırmaya genellikle gerekmez [5]. Bununla birlikte, `num_tpu_cores = 8` veya` tpu_cores = 8` [1] [5] gibi `eğitmenleri başlatırken kullanılacak TPU çekirdek sayısını belirtmelisiniz.
Alıntılar:[1] https://pytorch-lightning.readthedocs.io/en/0.7.6/tpu.html
[2] https://pytorch-lightning.readthedocs.io/en/0.7.6/introduction_guide.html
[3] https://lightning.ai/docs/pytorch/stable/notebooks/lightning_examples/mnist-tpu-training.html
[4] https://github.com/lightning-ai/pytorch-lightning/discussions/14743
[5] https://lightning.ai/docs/pytorch/1.5.9/advanced/tpu.html
[6] https://www.restack.io/p/pytorch-lightning-wring-concatity-matrix-cat-ai
[7] https://stackoverflow.com/questions/70136356/import-error-wayning-pytorch-lightning-project-on-colab-tpu
[8] https://stackoverflow.com/questions/66538407/unable-to-import-pytorch-glegning-on-google-colab