Skip to content

Instantly share code, notes, and snippets.

@samarthbhargav

samarthbhargav/CUDA Availability in Torch

Created September 24, 2018 10:00
Show Gist options
  • Save samarthbhargav/066d35449cd3f84102e610d17cd0c0e1 to your computer and use it in GitHub Desktop.
Save samarthbhargav/066d35449cd3f84102e610d17cd0c0e1 to your computer and use it in GitHub Desktop.
Download ZIP
One liner to test if cuda is available
Raw
CUDA Availability in Torch
python -c "import torch; print(torch.cuda.is_available())"
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment