Skip to content

Instantly share code, notes, and snippets.

@dmigo
Created January 7, 2022 11:15
Show Gist options
  • Save dmigo/3d4fddefa724f41118c89312656071b9 to your computer and use it in GitHub Desktop.
Save dmigo/3d4fddefa724f41118c89312656071b9 to your computer and use it in GitHub Desktop.
Checks if GPU is available on your machine
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