我已经阅读了许多关于如何理解 Tensorflow 是否在 GPU 上运行的问题和“指南”，但我仍然很困惑。
我截取了我的 session 截图，我想了解发生了什么，以及 Tensorflow 是在 GPU 还是 CPU 上运行。
最佳答案
罗伯托，试试这个
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
关于tensorflow - 检查 Tensorflow 是否在 GPU 上运行，我们在Stack Overflow上找到一个类似的问题： https://stackoverflow.com/questions/55691174/