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

目录:

一、问题解决

二、扩展内容

 

一、问题解决

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

posted @ 2020-04-12 16:07  细胞何  阅读(2795)  评论(0编辑  收藏  举报