Skip to content

Instantly share code, notes, and snippets.

@vaibhavkumar049

vaibhavkumar049/cudaex.py

Created May 25, 2019 03:24
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 vaibhavkumar049/d72a4e35ee8dfd9037f7ca8fe123e898 to your computer and use it in GitHub Desktop.
Save vaibhavkumar049/d72a4e35ee8dfd9037f7ca8fe123e898 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
cudaex.py
a = torch.ones(3, 2, device=cuda0)
b = torch.ones(3, 2, device=cuda0)
c = a + b
print(c)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment