shaneturner/checker.py

Created October 8, 2023 21:03

Star () You must be signed in to star a gist
Fork () You must be signed in to fork a gist

Learn more about clone URLs
Clone this repository at <script src="https://gist.github.com/shaneturner/c65a26c3a11e7f5f63f8076559a134fb.js"></script>
Save shaneturner/c65a26c3a11e7f5f63f8076559a134fb to your computer and use it in GitHub Desktop.

Download ZIP

Check python torch CUDA

Raw

	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