mberman84/checker.py

Created June 12, 2023 02:47

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/mberman84/abbe218942ad67b8adfb8f8f76d4abff.js"></script>
Save mberman84/abbe218942ad67b8adfb8f8f76d4abff to your computer and use it in GitHub Desktop.

Download ZIP

Cuda Checker

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