import tensorflow as tf
sess = tf.Session(config=tf.ConfigProto(log_device_placement=True))
查看日志信息若包含gpu信息,就是使用了gpu。
其他方法:跑计算量大的代码,通过 nvidia-smi 命令查看gpu的内存使用量。
本文介绍如何确认TensorFlow是否成功启用了GPU支持。通过配置Session并观察日志信息来判断,同时提供了一种运行计算密集型任务并通过nvidia-smi命令检查GPU负载的方法。
import tensorflow as tf
sess = tf.Session(config=tf.ConfigProto(log_device_placement=True))
查看日志信息若包含gpu信息,就是使用了gpu。
其他方法:跑计算量大的代码,通过 nvidia-smi 命令查看gpu的内存使用量。
2987

被折叠的 条评论
为什么被折叠?