search

pytorch / xla

Enabling PyTorch on XLA Devices (e.g. Google TPU)
https://pytorch.org/xla
Other
2.48k stars 480 forks source link

Is it possible to use XLA on TPU with BM1684 chip? #5289

Closed IzhanVarsky closed 1 year ago

IzhanVarsky commented 1 year ago

❓ Questions and Help

Hello, I am new to the world of TPU. I have this https://sophon.ai/product/introduce/se5-8.html micro server TPU with BM1684 chip. I tried to install pytorch xla to it (for python 3.8), but every .whl says that it "is not a supported wheel on this platform". Can I use pytorch xla on my TPU or not? If no then how can I find which devices are supported by pytorch xla?

JackCaoG commented 1 year ago

I am not aware of https://sophon.ai/product/introduce/se5-8.html, when we talking about TPU we usually means Google TPU. For non-goolgle workoad it is only available on Google Cloud. The issue you run into is just a python whl issue I think, is your machine is on python 3.8? That being said I don't think it will work given I felt like it is a different "TPU".

IzhanVarsky commented 1 year ago

Thank you for the response! Yes, I have python 3.8 on my TPU system, but I can't install pytorch xla wheels. So, am I right, that pytorch xla can be used only for Google TPUs but not for other TPU systems?

JackCaoG commented 1 year ago

yea, we only support Google TPU.