首页 > 其他 > 详细

(tensorflow计算)如何查看tensorflow计算用的是CPU还是GPU

时间:2020-04-12 16:26:33      阅读:618      评论:0      收藏:0      [点我收藏+]

目录:

一、问题解决

二、扩展内容

 

一、问题解决

sess.run()这行命令前面,加上如下内容:

  sess = tf.Session(config=tf.ConfigProto(allow_soft_placement=True, log_device_placement=True))

终端的Device mapping(设备映射)

技术分享图片

 

 

二、拓展内容

1、使用CPU却不使用GPU的问题解决

请参考: https://www.cnblogs.com/hutao722/p/9583214.html

 

 

 

本博客参考:

怎么知道tensorflow计算用的是CPU还是GPU                   https://blog.csdn.net/Arctic_Beacon/article/details/83536376

(tensorflow计算)如何查看tensorflow计算用的是CPU还是GPU

原文:https://www.cnblogs.com/hwh000/p/12685896.html

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!