Skip to content

Instantly share code, notes, and snippets.

@tchaton

tchaton/pl_1_4_tpu_cores_cmd_line.sh

Created July 20, 2021 15:11
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save tchaton/9f9500c82ae9ca3f023930585bbc35cf to your computer and use it in GitHub Desktop.
Save tchaton/9f9500c82ae9ca3f023930585bbc35cf to your computer and use it in GitHub Desktop.
Download ZIP
Raw
pl_1_4_tpu_cores_cmd_line.sh
export TPU_NAME=tpu-pod
python3 -m torch_xla.distributed.xla_dist --tpu=$TPU_NAME -- python3 script.py
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment