Rooba
/
lightning
mirror of https://github.com/Lightning-AI/lightning.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Update torch_xla installation instructions in tpu_basic.rst (#16865)

This commit is contained in:
Aditya Kane 2023-02-26 04:37:40 +05:30 committed by GitHub
parent 6b9ddf00dd
commit df6e37da1c
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
docs/source-pytorch/accelerators/tpu_basic.rst
View File

@ -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.