Skip to content

Instantly share code, notes, and snippets.

@shaneturner

shaneturner/checker.py

Created October 8, 2023 21:03
Show Gist options
  • Save shaneturner/c65a26c3a11e7f5f63f8076559a134fb to your computer and use it in GitHub Desktop.
Save shaneturner/c65a26c3a11e7f5f63f8076559a134fb to your computer and use it in GitHub Desktop.
Download ZIP
Check python torch CUDA
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