Skip to content

Instantly share code, notes, and snippets.

@kevin3
Created June 5, 2022 10:58
Show Gist options
  • Save kevin3/2cc6d8574e5fc11f58210e96508b13ca to your computer and use it in GitHub Desktop.
Save kevin3/2cc6d8574e5fc11f58210e96508b13ca to your computer and use it in GitHub Desktop.
testing if installed GPU enabled keras and tensorflow
###test if Using GPU for ML
#These commands in the Python terminal should bring up both a CPU and a GPU device, if TensorFlow correctly identifies your GPU.
import tensorflow as tf
tf.__version__
import keras
keras.__version__
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