(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


免责声明!

本站转载的文章为个人学习借鉴使用,本站对版权不负任何法律责任。如果侵犯了您的隐私权益,请联系本站邮箱yoyou2525@163.com删除。



 
粤ICP备18138465号  © 2018-2025 CODEPRJ.COM