Skip to content

Instantly share code, notes, and snippets.

@Infinitusvoid

Infinitusvoid/gist:76ec68a674e5aaf6e417523d5e1872d7

Created August 24, 2023 22:07
Show Gist options
  • Save Infinitusvoid/76ec68a674e5aaf6e417523d5e1872d7 to your computer and use it in GitHub Desktop.
Save Infinitusvoid/76ec68a674e5aaf6e417523d5e1872d7 to your computer and use it in GitHub Desktop.
Download ZIP
PyTorch : How to check if cuda is available ?
Raw
gistfile1.txt
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