fix tpu docs (#886)

2020-02-17 17:52:42 -05:00 · 2020-02-17 17:52:42 -05:00 · 9571de8757
parent 3562aa5aae
commit 9571de8757
1 changed files with 1 additions and 0 deletions
--- a/docs/source/tpu.rst
+++ b/docs/source/tpu.rst
@ -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:

 .. code-block:: bash
+
    $ python -m torch_xla.distributed.xla_dist
    --tpu=$TPU_POD_NAME
    --conda-env=torch-xla-nightly