From b2950296d59e825486c3e7a4c5b6e2b27f0c9965 Mon Sep 17 00:00:00 2001 From: William Falcon Date: Mon, 15 Feb 2021 08:58:15 -0500 Subject: [PATCH] fixed TPU docs (#5958) --- docs/source/advanced/tpu.rst | 10 ++++++++-- 1 file changed, 8 insertions(+), 2 deletions(-) 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+).