टेंसर फ्लो स्पष्ट डिवाइस आवश्यकता त्रुटि

Question

मैं एक gpu पर प्रशिक्षण कोड के साथ CIFAR10 ट्यूटोरियल चलाने और दूसरे पर eval कोड चलाने की कोशिश कर रहा हूं। मुझे यकीन है कि मेरे कंप्यूटर पर दो gpus हैं, और मैं यहां सरल उदाहरणों को चलाकर इसका परीक्षण कर सकता हूं: https://www.tensorflow.org/how_tos/using_gpu/index.html

हालांकि, with device('/gpu:0') का उपयोग करके CIFAR उदाहरण में अधिकांश चर के लिए काम नहीं करता है। मैंने जीपीयू बनाम सीपीयू, या एक या दूसरे पर सभी चरों पर विभिन्न चर के संयोजनों की पूरी कोशिश की। हमेशा कुछ चर, कुछ इस तरह के लिए एक ही त्रुटि:

Cannot assign a device to node 'shuffle_batch/random_shuffle_queue': Could not satisfy explicit device specification '/gpu:0' 

यह संभवतः टेन्सर प्रवाह में एक बग है या मैं कुछ याद आ रही है?

Answer 1

Could not satisfy explicit device specification का मतलब है कि आपके पास संबंधित डिवाइस नहीं है। क्या आपके पास वास्तव में आपकी मशीन पर एक CUDA- सक्षम GPU है?

अद्यतन: के रूप में यह नीचे चर्चा में निकला, यह त्रुटि भी अगर विशेष आपरेशन (इस मामले, RandomShuffleQueue में) GPU पर निष्पादित नहीं किया जा सकता है, क्योंकि यह केवल एक सीपीयू कार्यान्वयन है उठाया है।

आप TensorFlow आप के लिए एक उपकरण को चुनने (विशेष रूप से, सीपीयू कोई GPU कार्यान्वयन उपलब्ध है जब पर वापस गिरने) के साथ ठीक कर रहे हैं, के लिए, अपने विन्यास में allow_soft_placement सेट करने से प्रति this article के रूप में विचार करें:

sess = tf.Session(config=tf.ConfigProto(
    allow_soft_placement=True, log_device_placement=True))