< 返回技术文档列表

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

发布时间:2021-09-01 02:00:23⊙投诉举报

这篇文章将为大家详细讲解有关如何检测tensorflow是否使用gpu进行计算,小编觉得挺实用的,因此分享给大家做个参考,希望大家阅读完这篇文章后可以有所收获。

如下所示:

import tensorflow as tf
sess = tf.Session(config=tf.ConfigProto(log_device_placement=True))

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

其他方法:跑计算量大的代码,通过 nvidia-smi 命令查看gpu的内存使用量。

关于“如何检测tensorflow是否使用gpu进行计算”这篇文章就分享到这里了,希望以上内容可以对大家有一定的帮助,使各位可以学到更多知识,如果觉得文章不错,请把它分享出去让更多的人看到。


/template/Home/Zkeys/PC/Static