解決Tensorflow使用CPU報錯[ /job:localhost/replica:0/task:0/device:CPU:0 ]

最近在跟着<21個項目玩轉深度學習>中第三章的模型,但是遇見一個問題,我認為應該是CPU和GPU的使用問題.

解決方法是将slim檔案夾中的train_image_classifier.py檔案中的

tf.app.flags.DEFINE_boolean('clone_on_cpu', Flase,
                            'Use CPUs to deploy clones.')      

改為

tf.app.flags.DEFINE_boolean('clone_on_cpu', True,
                            'Use CPUs to deploy clones.')

      

修改完了 但是可以運作但是速度依舊慢得感人,對實驗室的電腦 已經無力吐槽.......