dmigo/cuda-availability.py

Created January 7, 2022 11:15

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

Download ZIP

Checks if GPU is available on your machine

Raw

cuda-availability.py

	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