Skip to content

Commit b295029

Browse files
fixed TPU docs (#5958)
1 parent 0a2fb05 commit b295029

File tree

1 file changed

+8
-2
lines changed

1 file changed

+8
-2
lines changed

docs/source/advanced/tpu.rst

Lines changed: 8 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -39,6 +39,12 @@ To access TPUs, there are three main ways.
3939

4040
----------------
4141

42+
Kaggle TPUs
43+
-----------
44+
For starting Kaggle projects with TPUs, refer to this `kernel <https://www.kaggle.com/pytorchlightning/pytorch-on-tpu-with-pytorch-lightning>`_.
45+
46+
---------
47+
4248
Colab TPUs
4349
----------
4450
Colab is like a jupyter notebook with a free GPU or TPU
@@ -58,8 +64,8 @@ To get a TPU on colab, follow these steps:
5864

5965
.. code-block::
6066
61-
!curl https://raw.githubusercontent.com/pytorch/xla/master/contrib/scripts/env-setup.py -o pytorch-xla-env-setup.py
62-
!python pytorch-xla-env-setup.py --version nightly --apt-packages libomp5 libopenblas-dev
67+
!curl https://raw.githubusercontent.com/pytorch/xla/master/contrib/scripts/env-setup.py -o pytorch-xla-env-setup.py
68+
!python pytorch-xla-env-setup.py --version 1.7 --apt-packages libomp5 libopenblas-dev
6369
6470
5. Once the above is done, install PyTorch Lightning (v 0.7.0+).
6571

0 commit comments

Comments
 (0)