問題內(nèi)容如下

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運(yùn)行，需要在CPU運(yùn)行
解決辦法：修改配置里面的allow_soft_placement=True，而allow_soft_placement=True，就是運(yùn)行在出現(xiàn)不允許在GPU運(yùn)行的時(shí)候，可以切換到CPU運(yùn)行。

【2017/8/9】20：20
經(jīng)過一個(gè)下午的不斷的調(diào)試認(rèn)證，發(fā)現(xiàn)，allow_soft_placement=True，就是在GPU跑不動(dòng)程序的時(shí)候，自動(dòng)切換到CPU運(yùn)行，這種方式雖然能夠確保程序的正常運(yùn)行，但是卻不能保證GPU出現(xiàn)問題能夠及時(shí)的反饋，故而還是推薦使用allow_soft_placement=True的選項(xiàng)。

創(chuàng)作挑戰(zhàn)賽新人創(chuàng)作獎(jiǎng)勵(lì)來咯，堅(jiān)持創(chuàng)作打卡瓜分現(xiàn)金大獎(jiǎng)

總結(jié)

以上是生活随笔為你收集整理的【Ubuntu-Tensorflow】InvalidArgumentError GPU不能使用的问题的全部?jī)?nèi)容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯(cuò)，歡迎將生活随笔推薦給好友。