Tensorflow: Tensorflow n'attribue aucune tâche au GPU dans l'exemple cifar10
J'exécute le script cifar10_multi_gpu_train.py avec la journalisation de placement de périphérique activée, et je vois que toutes les opérations sont placées sur le CPU. Quand je l'exécute, il affiche :
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
.....
.....
J'imagine que la ligne Device mapping: no known devices. est censée répertorier les périphériques associés à la session en cours, mais pourquoi n'a-t-elle aucun périphérique ? Le script appelle
tf.device('/gpu:0') .
Mon GPU est Nvidia GeForce GTX 970.
jricheimer
Tous les 3 commentaires
Résolu cela.
J'ai dû construire accidentellement le package python sans le drapeau "--config=cuda". Lorsque j'ai reconstruit avec ce drapeau, puis pip a désinstallé tensorflow, puis réinstallé avec le paquet nouvellement construit, cela a parfaitement fonctionné.
jricheimer
le 23 nov. 2015
Le même problème s'est produit lorsque j'exécutais le script cifar10_multi_gpu_train.py avec la journalisation du placement de l'appareil activée. Je n'ai aucune idée de ce que vous voulez dire lorsque vous dites "J'ai dû accidentellement construire le package python sans le drapeau "--config=cuda"", pouvez-vous dire quel package python vous avez reconstruit?
GuangmingZhu
le 31 oct. 2016
@GuangmingZhu , sur la page de téléchargement et de configuration de TensorFlow, la commande permettant de créer le package pip doit être :
bazel build -c opt --config=cuda //tensorflow/tools/pip_package:build_pip_package
J'ai juste oublié le --config=cuda .
jricheimer
le 1 nov. 2016
Questions connexes
ppwwyyxx
·
3Commentaires
wfs
·
3Commentaires
nicholaslocascio
·
3Commentaires
aerointern16
·
3Commentaires
davidbernat
·
3Commentaires
Commentaire le plus utile
Résolu cela.
J'ai dû construire accidentellement le package python sans le drapeau "--config=cuda". Lorsque j'ai reconstruit avec ce drapeau, puis pip a désinstallé tensorflow, puis réinstallé avec le paquet nouvellement construit, cela a parfaitement fonctionné.