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

来源:互联网 发布:vip域名备案 编辑:程序博客网 时间:2024/05/22 00:40
import tensorflow as tfsess = tf.Session(config=tf.ConfigProto(log_device_placement=True))

查看日志信息若包含gpu信息,就是使用了gpu。
其他方法:跑计算量大的代码,通过 nvidia-smi 命令查看gpu的内存使用量。

原创粉丝点击