tensorflow - 检查 Tensorflow 是否在 GPU 上运行

Question

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

我截取了我的 session 截图，我想了解发生了什么，以及 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