問題內容如下

InvalidArgumentError (see above for traceback): Cannot assign a device to node 'train/final/fc3/b/Momentum': Could not satisfy explicit device specification '/device:GPU:0' because no devices matching that specification are registered in this process; available devices: /job:localhost/replica:0/task:0/cpu:0 Colocation Debug Info: Colocation group had the following types and devices: ApplyMomentum: CPU Mul: CPU Sum: CPU Abs: CPU Const: CPU Assign: CPU Identity: CPU VariableV2: CPU [[Node: train/final/fc3/b/Momentum = VariableV2[_class=["loc:@train/final/fc3/b"], container="", dtype=DT_FLOAT, shape=[29], shared_name="", _device="/device:GPU:0"]()]]

問題原因：有些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不能使用的问题的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。