aurora1625/check tf cuda.py

Last active March 27, 2022 23:22

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

Download ZIP

Check #tensorflow #CUDA

Raw

check tf cuda.py

	import tensorflow as tf
	print(tf.test.is_built_with_cuda())

	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