I have read many questions and "guides" on how to understand if Tensorflow is running on GPU but I am still quite confused.
我已经截取了会话的屏幕截图,我想了解发生了什么,以及 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
本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系:hwhale#tublm.com(使用前将#替换为@)