Publish Python 3.11 Torch XLA wheels

pytorch / xla

Enabling PyTorch on XLA Devices (e.g. Google TPU)

https://pytorch.org/xla

Other

2.49k stars 480 forks source link

Publish Python 3.11 Torch XLA wheels #8250

Open metrizable opened 1 month ago

metrizable commented 1 month ago

🚀 The feature, motivation and pitch

For users that wish to run Torch on XLA devices in Python 3.11, having pre-built Torch XLA wheels will provide a smooth, reliable installation experience.

Alternatives

Users may build their own wheels, but this may result in issue generation. As well, building once for many uses saves general compute burn.

Additional context

No response

cc @bdhirsh

malfet commented 1 month ago

Moving the issue to xla repository, but @JackCaoG do not hesitate to move it back if you believe it belongs with pytorch

JackCaoG commented 1 month ago

we actually do have python 3.11 whl, but we didn't mention it in https://github.com/pytorch/xla#available-docker-images-and-wheels

for the nightly, if you replace the cp310-cp310 to cp311-cp311 you can get the 3.11 wheel. We will also publish the 3.11 wheel for the upcoming release.