darcwader/test_tf_gpu.py

Created May 30, 2018 12:56

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

Download ZIP

test tensorflow GPU

Raw

test_tf_gpu.py

	from tensorflow.python.client import device_lib
	print(device_lib.list_local_devices())

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