Skip to content

Instantly share code, notes, and snippets.

@AngelBerihuete

AngelBerihuete/check_CPU_or_GPU_runing.py

Created June 17, 2019 11:00
Show Gist options
  • Save AngelBerihuete/7894f15d78a7f8b4355566213d307dfd to your computer and use it in GitHub Desktop.
Save AngelBerihuete/7894f15d78a7f8b4355566213d307dfd to your computer and use it in GitHub Desktop.
Download ZIP
Raw
check_CPU_or_GPU_runing.py
# from https://www.quora.com/How-can-I-ensure-that-my-TensorFlow-is-running-by-GPU-or-not-by-GPU
sess = tf.Session(config=tf.ConfigProto(log_device_placement=True))
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment