【Ubuntu-Tensorflow】InvalidArgumentError GPU不能使用的问题
生活随笔
收集整理的這篇文章主要介紹了
【Ubuntu-Tensorflow】InvalidArgumentError GPU不能使用的问题
小編覺得挺不錯的,現在分享給大家,幫大家做個參考.
問題內容如下
問題原因:有些tf.Variable()不允許在GPU運行,需要在CPU運行
解決辦法:修改配置里面的allow_soft_placement=True,而allow_soft_placement=True,就是運行在出現不允許在GPU運行的時候,可以切換到CPU運行。
【2017/8/9】20:20
經過一個下午的不斷的調試認證,發現,allow_soft_placement=True,就是在GPU跑不動程序的時候,自動切換到CPU運行,這種方式雖然能夠確保程序的正常運行,但是卻不能保證GPU出現問題能夠及時的反饋,故而還是推薦使用allow_soft_placement=True的選項。
總結
以上是生活随笔為你收集整理的【Ubuntu-Tensorflow】InvalidArgumentError GPU不能使用的问题的全部內容,希望文章能夠幫你解決所遇到的問題。
- 上一篇: 专家:年轻人工资低是因为能力不够 很少有
- 下一篇: Intel显卡兵分两路!独显、集显大不同