Skip to content

Instantly share code, notes, and snippets.

@engividal
Created February 29, 2020 16:12
Show Gist options
  • Save engividal/b3fe6d259703e1660acfeb9d8c40e828 to your computer and use it in GitHub Desktop.
Save engividal/b3fe6d259703e1660acfeb9d8c40e828 to your computer and use it in GitHub Desktop.
This code check whether GPU is being or not
Display the source blob
Display the rendered blob
Raw
# Check whether GPU is being or not
from tensorflow.python.client import device_lib
print(device_lib.list_local_devices())
tf_device="GPU:0"
!nvidia-smi
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment