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

时间:2020-04-12 16:26:33   收藏:0   阅读:624

目录:

一、问题解决

二、扩展内容

 

一、问题解决

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

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

评论(0
© 2014 bubuko.com 版权所有 - 联系我们:wmxa8@hotmail.com
打开技术之扣,分享程序人生!