Skip to content

Instantly share code, notes, and snippets.

@mberman84
Created June 12, 2023 02:47
Show Gist options
  • Save mberman84/abbe218942ad67b8adfb8f8f76d4abff to your computer and use it in GitHub Desktop.
Save mberman84/abbe218942ad67b8adfb8f8f76d4abff to your computer and use it in GitHub Desktop.
Cuda Checker
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