Created
February 29, 2020 16:12
-
-
Save engividal/b3fe6d259703e1660acfeb9d8c40e828 to your computer and use it in GitHub Desktop.
This code check whether GPU is being or not
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
# 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