logo资料库

检测tensorflow是否使用gpu进行计算的方式.pdf

第1页 / 共1页
资料共1页,全文预览结束
检测检测tensorflow是否使用 今天小编就为大家分享一篇检测tensorflow是否使用gpu进行计算的方式,具有很好的参考价值,希望对大家有 所帮助。一起跟随小编过来看看吧 是否使用gpu进行计算的方式 进行计算的方式 如下所示: 如下所示: import tensorflow as tf sess = tf.Session(config=tf.ConfigProto(log_device_placement=True)) 查看日志信息若包含gpu信息,就是使用了gpu。 其他方法:跑计算量大的代码,通过 nvidia-smi 命令查看gpu的内存使用量。 其他方法: 以上这篇检测tensorflow是否使用gpu进行计算的方式就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大 家多多支持我们。
分享到:
收藏