Failed to install xla gpu

pytorch / xla

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

https://pytorch.org/xla

Other

2.39k stars 429 forks source link

Failed to install xla gpu #7634

Open Beakboomboom opened 3 weeks ago

Beakboomboom commented 3 weeks ago

❓ Questions and Help

pip install torch_xla-2.2.0-cp310-cp310-manylinux_2_28_x86_64.whl But got the error: ERROR: torch_xla-2.2.0-cp310-cp310-manylinux_2_28_x86_64.whl is not a supported wheel on this platform.

How can i install torch_xla on GPU ?

miladm commented 2 weeks ago

@Beakboomboom

did you try https://storage.googleapis.com/pytorch-xla-releases/wheels/cuda/12.1/torch_xla-2.2.0-cp310-cp310-manylinux_2_28_x86_64.whl from the GPU whl installation instructions on the landing page (https://github.com/pytorch/xla). Let us know if this works for you please.