diff --git a/docs/source/advanced/tpu.rst b/docs/source/advanced/tpu.rst index 549a3a1cd25d6..e98003a3edab6 100644 --- a/docs/source/advanced/tpu.rst +++ b/docs/source/advanced/tpu.rst @@ -39,6 +39,12 @@ To access TPUs, there are three main ways. ---------------- +Kaggle TPUs +----------- +For starting Kaggle projects with TPUs, refer to this `kernel `_. + +--------- + Colab TPUs ---------- 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: .. code-block:: - !curl https://raw.githubusercontent.com/pytorch/xla/master/contrib/scripts/env-setup.py -o pytorch-xla-env-setup.py - !python pytorch-xla-env-setup.py --version nightly --apt-packages libomp5 libopenblas-dev + !curl https://raw.githubusercontent.com/pytorch/xla/master/contrib/scripts/env-setup.py -o pytorch-xla-env-setup.py + !python pytorch-xla-env-setup.py --version 1.7 --apt-packages libomp5 libopenblas-dev 5. Once the above is done, install PyTorch Lightning (v 0.7.0+).