iamaaditya/test_cudnn.py

Created August 30, 2018 18:19

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

Download ZIP

test cudnn

Raw

test_cudnn.py

	import torch
	assert torch.backends.cudnn.enabled == True
	print(torch.backends.cudnn.version())

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