ThoenigAdrian/tensorflow_gpu_support_test.py

Created December 9, 2019 16: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/ThoenigAdrian/e62970d6167074e7e6fc620ec4fc9847.js"></script>
Save ThoenigAdrian/e62970d6167074e7e6fc620ec4fc9847 to your computer and use it in GitHub Desktop.

Download ZIP

Checks if gpu is supported after pip install tensorflow-gpu

Raw

tensorflow_gpu_support_test.py

	import tensorflow
	tensorflow.Session(config=tensorflow.ConfigProto(log_device_placement=True))

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