Skip to content

Instantly share code, notes, and snippets.

@deeperlearner

deeperlearner/tf_gpu_test.py

Created July 7, 2021 12:47
Show Gist options
  • Save deeperlearner/910ee60445d91cf17b374cf539e251f4 to your computer and use it in GitHub Desktop.
Save deeperlearner/910ee60445d91cf17b374cf539e251f4 to your computer and use it in GitHub Desktop.
Download ZIP
tf_gpu_test.py
Raw
tf_gpu_test.py
import tensorflow as tf
if tf.test.gpu_device_name():
print(f"Default GPU Device: {tf.test.gpu_device_name()}")
else:
print("Please install GPU version of TF")
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment