Add link to TPU Pods tutorial. (#2477)

This commit is contained in:
zcain117 2020-07-02 21:57:17 -07:00 committed by GitHub
parent 020c332ae9
commit 6d9c7bf0b0
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 0 deletions

View File

@ -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