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

如下所示：

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

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

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

来源：https://blog.csdn.net/castle_cc/article/details/78389082