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

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

如下所示：

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

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

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

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