您的位置:首页 > 其它

检测tensorflow是否使用gpu进行计算

2017-10-29 23:10 991 查看
import tensorflow as tf
sess = tf.Session(config=tf.ConfigProto(log_device_placement=True))


查看日志信息若包含gpu信息,就是使用了gpu。

其他方法:跑计算量大的代码,通过 nvidia-smi 命令查看gpu的内存使用量。
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签:  tensorflow gpu
相关文章推荐