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

fix tpu docs (#886)

This commit is contained in:
William Falcon 2020-02-17 17:52:42 -05:00 committed by GitHub
parent 3562aa5aae
commit 9571de8757
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
docs/source/tpu.rst
View File

@ -147,6 +147,7 @@ To train on more than 8 cores, your code actually doesn't change!
All you need to do is submit the following command: All you need to do is submit the following command:
.. code-block:: bash .. code-block:: bash
$ python -m torch_xla.distributed.xla_dist $ python -m torch_xla.distributed.xla_dist
--tpu=$TPU_POD_NAME --tpu=$TPU_POD_NAME
--conda-env=torch-xla-nightly --conda-env=torch-xla-nightly