vsay01/check_cuda.py

Created January 22, 2020 01:43

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

Select an option

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

Download ZIP

check if GPU available

Raw

check_cuda.py

	# check if CUDA is available
	use_cuda = torch.cuda.is_available()

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment