Skip to content

Instantly share code, notes, and snippets.

@ZekunZh

ZekunZh/validate_pytorch_installation_with_gpu.md

Last active December 10, 2021 16:32
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 ZekunZh/5b83b295b217ed876d06b3445eb1f064 to your computer and use it in GitHub Desktop.
Save ZekunZh/5b83b295b217ed876d06b3445eb1f064 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
validate_pytorch_installation_with_gpu.md 
python3 -c "import torch; print(torch.tensor([0.], device='cuda:0'))"
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment