检查Tensorflow是否在GPU上运行

Question

我已经阅读了很多有关如何理解Tensorflow是否在GPU上运行的问题和“指南”，但是我仍然很困惑。

我已经为我的会话截图了，我想了解发生了什么，以及Tensorflow是否在GPU或CPU上运行。

Answer 1

罗伯托，试试这个

import tensorflow as tf
if tf.test.gpu_device_name():
    print('Default GPU Device: {}'.format(tf.test.gpu_device_name()))
else:
    print("Please install GPU version of TF")

输出应为

Default GPU Device: /device:GPU:0

GPU 0是您的GTX 860m