@@ -24,21 +24,21 @@ started:
2424### TPU
2525
2626To install PyTorch/XLA stable build in a new TPU VM:
27- Note: Builds are available for Python 3.8 to 3.11 ; please use one of the supported versions.
27+ Note: Builds are available for Python 3.10 to 3.13 ; please use one of the supported versions.
2828
2929``` sh
3030# - for venv
3131# python3.11 -m venv py311
3232# - for conda
3333# conda create -n py311 python=3.11
3434
35- pip install torch==2.8 .0 ' torch_xla[tpu]==2.8 .0'
35+ pip install torch==2.9 .0 ' torch_xla[tpu]==2.9 .0'
3636
3737# Optional: if you're using custom kernels, install pallas dependencies
3838pip install --pre torch_xla[pallas] --index-url https://us-python.pkg.dev/ml-oss-artifacts-published/jax/simple/ --find-links https://storage.googleapis.com/jax-releases/libtpu_releases.html
3939```
4040** As of 07/16/2025 and starting from Pytorch/XLA 2.8 release, PyTorch/XLA will
41- provide nightly and release wheels for Python 3.11 to 3.13**
41+ provide nightly and release wheels for Python 3.10 to 3.13**
4242To install PyTorch/XLA nightly build in a new TPU VM:
4343
4444``` sh
@@ -280,29 +280,28 @@ The torch wheel version `2.9.0.dev20250423+cpu` can be found at https://download
280280
281281| Version | Cloud TPU VMs Wheel |
282282| ---------| -------------------|
283+ | 2.9 (Python 3.10) | ` https://storage.googleapis.com/pytorch-xla-releases/wheels/tpuvm/torch_xla-2.9.0-cp310-cp310-manylinux_2_28_x86_64.whl ` |
284+ | 2.8 (Python 3.10) | ` https://storage.googleapis.com/pytorch-xla-releases/wheels/tpuvm/torch_xla-2.8.0-cp310-cp310-manylinux_2_28_x86_64.whl ` |
283285| 2.7 (Python 3.10) | ` https://storage.googleapis.com/pytorch-xla-releases/wheels/tpuvm/torch_xla-2.7.0-cp310-cp310-manylinux_2_28_x86_64.whl ` |
284286| 2.6 (Python 3.10) | ` https://storage.googleapis.com/pytorch-xla-releases/wheels/tpuvm/torch_xla-2.6.0-cp310-cp310-manylinux_2_28_x86_64.whl ` |
285287| 2.5 (Python 3.10) | ` https://storage.googleapis.com/pytorch-xla-releases/wheels/tpuvm/torch_xla-2.5.0-cp310-cp310-manylinux_2_28_x86_64.whl ` |
286288| 2.4 (Python 3.10) | ` https://storage.googleapis.com/pytorch-xla-releases/wheels/tpuvm/torch_xla-2.4.0-cp310-cp310-manylinux_2_28_x86_64.whl ` |
287289| 2.3 (Python 3.10) | ` https://storage.googleapis.com/pytorch-xla-releases/wheels/tpuvm/torch_xla-2.3.0-cp310-cp310-manylinux_2_28_x86_64.whl ` |
288- | 2.2 (Python 3.10) | ` https://storage.googleapis.com/pytorch-xla-releases/wheels/tpuvm/torch_xla-2.2.0-cp310-cp310-manylinux_2_28_x86_64.whl ` |
289- | 2.1 (XRT + Python 3.10) | ` https://storage.googleapis.com/pytorch-xla-releases/wheels/xrt/tpuvm/torch_xla-2.1.0%2Bxrt-cp310-cp310-manylinux_2_28_x86_64.whl ` |
290- | 2.1 (Python 3.8) | ` https://storage.googleapis.com/pytorch-xla-releases/wheels/tpuvm/torch_xla-2.1.0-cp38-cp38-linux_x86_64.whl ` |
291290
292291</details >
293292
294293### Docker
295294NOTE: Since PyTorch/XLA 2.7, all builds will use the C++11 ABI by default
296295| Version | Cloud TPU VMs Docker |
297296| --- | ----------- |
297+ | 2.9 | ` us-central1-docker.pkg.dev/tpu-pytorch-releases/docker/xla:r2.9.0_3.10_tpuvm ` |
298+ | 2.8 | ` us-central1-docker.pkg.dev/tpu-pytorch-releases/docker/xla:r2.8.0_3.10_tpuvm ` |
298299| 2.7 | ` us-central1-docker.pkg.dev/tpu-pytorch-releases/docker/xla:r2.7.0_3.10_tpuvm ` |
299300| 2.6 | ` us-central1-docker.pkg.dev/tpu-pytorch-releases/docker/xla:r2.6.0_3.10_tpuvm ` |
300301| 2.6 (C++11 ABI) | ` us-central1-docker.pkg.dev/tpu-pytorch-releases/docker/xla:r2.6.0_3.10_tpuvm_cxx11 ` |
301302| 2.5 | ` us-central1-docker.pkg.dev/tpu-pytorch-releases/docker/xla:r2.5.0_3.10_tpuvm ` |
302303| 2.4 | ` us-central1-docker.pkg.dev/tpu-pytorch-releases/docker/xla:r2.4.0_3.10_tpuvm ` |
303304| 2.3 | ` us-central1-docker.pkg.dev/tpu-pytorch-releases/docker/xla:r2.3.0_3.10_tpuvm ` |
304- | 2.2 | ` us-central1-docker.pkg.dev/tpu-pytorch-releases/docker/xla:r2.2.0_3.10_tpuvm ` |
305- | 2.1 | ` us-central1-docker.pkg.dev/tpu-pytorch-releases/docker/xla:r2.1.0_3.10_tpuvm ` |
306305| nightly python | ` us-central1-docker.pkg.dev/tpu-pytorch-releases/docker/xla:nightly_3.10_tpuvm ` |
307306
308307To use the above dockers, please pass ` --privileged --net host --shm-size=16G ` along. Here is an example:
0 commit comments