Skip to content

Instantly share code, notes, and snippets.

@wookim3

wookim3/TPU.py

Created March 8, 2020 00:01
Show Gist options
  • Save wookim3/5a02ed6eb2b3fac30f6c11f722ae1a29 to your computer and use it in GitHub Desktop.
Save wookim3/5a02ed6eb2b3fac30f6c11f722ae1a29 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
TPU.py
VERSION = "20200220" #@param ["20200220","nightly", "xrt==1.15.0"]
!curl https://raw.githubusercontent.com/pytorch/xla/master/contrib/scripts/env-setup.py -o pytorch-xla-env-setup.py
!python pytorch-xla-env-setup.py --version $VERSION
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment