Tensorflow: لا يقوم Tensorflow بتعيين أي مهام إلى GPU في مثال cifar10

تم إنشاؤها على ٢٠ نوفمبر ٢٠١٥  ·  3تعليقات  ·  مصدر: tensorflow/tensorflow

أنا أقوم بتشغيل البرنامج النصي cifar10_multi_gpu_train.py مع تشغيل تسجيل موضع الجهاز ، وأرى أن جميع العمليات يتم وضعها على وحدة المعالجة المركزية. عندما أقوم بتشغيله ، فإنه ينتج:

Filling queue with 20000 CIFAR images before starting to train. This will take a few minutes.
I tensorflow/core/common_runtime/local_device.cc:25] Local device intra op parallelism threads: 8
I tensorflow/core/common_runtime/direct_session.cc:45] Direct session inter op parallelism threads: 8
Device mapping: no known devices.
I tensorflow/core/common_runtime/direct_session.cc:111] Device mapping:

softmax_linear/biases/ExponentialMovingAverage: /job:localhost/replica:0/task:0/cpu:0
I tensorflow/core/common_runtime/simple_placer.cc:289] softmax_linear/biases/ExponentialMovingAverage: /job:localhost/replica:0/task:0/cpu:0
.....
.....

أتخيل أن السطر Device mapping: no known devices. من المفترض أن يسرد الأجهزة المرتبطة بالجلسة الحالية ، لكن لماذا لا يحتوي على أي أجهزة؟ يستدعي البرنامج النصي
tf.device('/gpu:0') .

وحدة معالجة الرسومات الخاصة بي هي Nvidia GeForce GTX 970.

التعليق الأكثر فائدة

حل هذا.

لابد أنني قمت ببناء حزمة python عن طريق الخطأ بدون علامة "--config = cuda". عندما أعدت البناء باستخدام هذه العلامة ، ثم تم إلغاء تثبيت tensorflow ، ثم إعادة تثبيته مع الحزمة المبنية حديثًا ، فقد عملت بشكل مثالي.

ال 3 كومينتر

حل هذا.

لابد أنني قمت ببناء حزمة python عن طريق الخطأ بدون علامة "--config = cuda". عندما أعدت البناء باستخدام هذه العلامة ، ثم تم إلغاء تثبيت tensorflow ، ثم إعادة تثبيته مع الحزمة المبنية حديثًا ، فقد عملت بشكل مثالي.

حدثت نفس المشكلة عندما كنت أقوم بتشغيل البرنامج النصي cifar10_multi_gpu_train.py مع تشغيل تسجيل موضع الجهاز. ليس لدي أي فكرة عما تقصده عندما تقول "لابد أني قمت عن طريق الخطأ ببناء حزمة python بدون" --config = cuda "flag" ، هل يمكنك تحديد حزمة python التي قمت بإعادة بنائها؟

GuangmingZhu ، في صفحة تنزيل وإعداد TensorFlow ، يجب أن يكون الأمر الخاص ببناء حزمة النقطة كما يلي:

bazel build -c opt --config=cuda //tensorflow/tools/pip_package:build_pip_package

لقد نسيت للتو --config=cuda .

هل كانت هذه الصفحة مفيدة؟
0 / 5 - 0 التقييمات