Tensorflow: Tensorflow weist GPU im cifar10-Beispiel keine Aufgaben zu
Ich führe das cifar10_multi_gpu_train.py -Skript mit aktivierter Geräteplatzierungsprotokollierung aus und sehe, dass alle Operationen auf der CPU platziert werden. Wenn ich es ausführe, gibt es aus:
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
.....
.....
Ich würde mir vorstellen, dass die Zeile Device mapping: no known devices. die Geräte auflisten soll, die der aktuellen Sitzung zugeordnet sind, aber warum hat sie keine Geräte? Das Skript ruft
tf.device('/gpu:0') .
Meine GPU ist Nvidia GeForce GTX 970.
jricheimer
Alle 3 Kommentare
Gelöst.
Ich muss das Python-Paket versehentlich ohne das Flag "--config=cuda" erstellt haben. Als ich mit diesem Flag neu erstellte, dann pip deinstallierte Tensorflow und dann mit dem neu erstellten Paket neu installierte, funktionierte es perfekt.
jricheimer
am 23. Nov. 2015
Das gleiche Problem trat auf, als ich das Skript cifar10_multi_gpu_train.py mit aktivierter Protokollierung der Geräteplatzierung ausführte. Ich habe keine Ahnung, was Sie meinen, wenn Sie sagen: "Ich muss das Python-Paket versehentlich ohne das Flag "--config=cuda" erstellt haben", können Sie sagen, welches Python-Paket Sie neu erstellt haben?
GuangmingZhu
am 31. Okt. 2016
@GuangmingZhu , auf der TensorFlow -Download- und Einrichtungsseite sollte der Befehl zum Erstellen des Pip-Pakets lauten:
bazel build -c opt --config=cuda //tensorflow/tools/pip_package:build_pip_package
Ich habe nur die --config=cuda vergessen.
jricheimer
am 1. Nov. 2016
Verwandte Themen
aerointern16
·
3Kommentare
indiejoseph
·
3Kommentare
myme5261314
·
3Kommentare
ilblackdragon
·
3Kommentare
3rduncle
·
3Kommentare
Hilfreichster Kommentar
Gelöst.
Ich muss das Python-Paket versehentlich ohne das Flag "--config=cuda" erstellt haben. Als ich mit diesem Flag neu erstellte, dann pip deinstallierte Tensorflow und dann mit dem neu erstellten Paket neu installierte, funktionierte es perfekt.