Tensorflow錯誤匯總與解決辦法

 

 

2017-12-19

盤點遇到的各種Tensorflow坑（此博客不定期更新）：

１．InvalidArgumentError (see above for traceback): Cannot assign a device for operation 'shuffle_batch': Could not satisfy explicit device specification '/device:GPU:0' because no supported kernel for GPU devices is available.

各種google之后，終於找到了解決辦法：

原文鏈接為：https://github.com/tensorflow/tensorflow/issues/2285

於是，把代碼中的“with tf.device('/gpu:0'):”改為了“with tf.device(None):”，問題終於解決了。