Skip to content

Instantly share code, notes, and snippets.

@yaroslavvb2

yaroslavvb2/transfers.py

Created October 29, 2017 02:03
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 yaroslavvb2/3f2cd20df7583bee899f101fe89c77d0 to your computer and use it in GitHub Desktop.
Save yaroslavvb2/3f2cd20df7583bee899f101fe89c77d0 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
transfers.py
a = a.gpu() # copies tensor to default GPU (GPU0)
a = a.gpu(0) # copies tensor to GPU0
a = a.gpu(1) # copies tensor to GPU1
a = a.cpu() # copies tensor back to CPU
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment