码迷,mamicode.com
首页 > 其他好文 > 详细

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

时间:2020-12-17 12:54:30      阅读:3      评论:0      收藏:0      [点我收藏+]

标签:代码   tensor   命令   smi   其他   href   ESS   通过   tar   

更多python教程请到: 菜鸟教程www.piaodoo.com

人人影视www.sfkyty.com


如下所示:

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

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

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

以上这篇检测tensorflow是否使用gpu进行计算的方式就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持菜鸟教程www.piaodoo.com。

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

标签:代码   tensor   命令   smi   其他   href   ESS   通过   tar   

原文地址:https://www.cnblogs.com/piaodoo/p/14124982.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!