Skip to content

Commit

Permalink
Add link to TPU Pods tutorial. (#2477)
Browse files Browse the repository at this point in the history
  • Loading branch information
zcain117 authored Jul 3, 2020
1 parent 020c332 commit 6d9c7bf
Showing 1 changed file with 3 additions and 0 deletions.
3 changes: 3 additions & 0 deletions docs/source/tpu.rst
Original file line number Diff line number Diff line change
Expand Up @@ -153,6 +153,9 @@ All you need to do is submit the following command:
--conda-env=torch-xla-nightly
-- python /usr/share/torch-xla-0.5/pytorch/xla/test/test_train_imagenet.py --fake_data
See `this guide <https://cloud.google.com/tpu/docs/tutorials/pytorch-pod>`_
on how to set up the instance groups and VMs needed to run TPU Pods.

----------------

16 bit precision
Expand Down

0 comments on commit 6d9c7bf

Please sign in to comment.