(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