Skip to content

Instantly share code, notes, and snippets.

@garystafford

garystafford/print_pytorch_gpu_versions.py

Last active December 3, 2023 22:45
Show Gist options
  • Save garystafford/3c6f1791a71bde41c7f5fd0ade73fefb to your computer and use it in GitHub Desktop.
Save garystafford/3c6f1791a71bde41c7f5fd0ade73fefb to your computer and use it in GitHub Desktop.
Download ZIP
Raw
print_pytorch_gpu_versions.py
import torch
print(torch.__version__) # e.g., 2.0.0 (at the time of the post)
print(torch.cuda.get_device_name(0)) # e.g., NVIDIA A10G
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment