(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