From df6e37da1c8acf1934dcf2a108e7d355b0709b70 Mon Sep 17 00:00:00 2001 From: Aditya Kane <64411306+AdityaKane2001@users.noreply.github.com> Date: Sun, 26 Feb 2023 04:37:40 +0530 Subject: [PATCH] Update torch_xla installation instructions in tpu_basic.rst (#16865) --- docs/source-pytorch/accelerators/tpu_basic.rst | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/docs/source-pytorch/accelerators/tpu_basic.rst b/docs/source-pytorch/accelerators/tpu_basic.rst index 2e0d764148..0b6118db4e 100644 --- a/docs/source-pytorch/accelerators/tpu_basic.rst +++ b/docs/source-pytorch/accelerators/tpu_basic.rst @@ -78,7 +78,7 @@ To get a TPU on colab, follow these steps: .. code-block:: - !pip install cloud-tpu-client https://storage.googleapis.com/tpu-pytorch/wheels/torch_xla-1.12-cp39-cp39m-linux_x86_64.whl + !pip install cloud-tpu-client https://storage.googleapis.com/tpu-pytorch/wheels/torch_xla-1.13-cp38-cp38m-linux_x86_64.whl 5. Once the above is done, install PyTorch Lightning.