Skip to content

Instantly share code, notes, and snippets.

@mberman84

mberman84/checker.py

Created June 12, 2023 02:47
Show Gist options
  • Star 5 You must be signed in to star a gist
  • Fork 4 You must be signed in to fork a gist
  • Save mberman84/abbe218942ad67b8adfb8f8f76d4abff to your computer and use it in GitHub Desktop.
Save mberman84/abbe218942ad67b8adfb8f8f76d4abff to your computer and use it in GitHub Desktop.
Download ZIP
Cuda Checker
Raw
checker.py
import torch
print(torch.version.cuda)
print(torch.cuda.is_available())
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment