Tensorflow-Dokumentation enthält den folgenden Beispielcode zum Ermitteln der Geräteplatzierung von Knoten. Das heißt, auf welchem Gerät eine bestimmte Berechnung stattfindet.
%Vor%Für mich wird der Code die Standorte der Geräte nicht wie gewünscht ausdrucken. Ich benutze das Jupyter-Notebook, das auf Ubuntu läuft. Wie kann ich das beheben oder die Informationen auf andere Weise herausfinden?
Für Jupyter (und andere) Benutzer gibt es eine kürzlich hinzugefügte Funktion, die es ermöglicht, die Geräteplatzierung zurückzulesen, wenn Sie eine Session.run() rufen Sie an und drucken Sie es in Ihr Notizbuch.
Das metadata.partition_graphs enthält die tatsächlichen Knoten des ausgeführten Graphen, partitioniert nach Gerät. Die Partitionen sind nicht explizit mit dem Gerät gekennzeichnet, das sie darstellen, aber für jedes NodeDef im Diagramm ist das Feld device festgelegt.
Leider sieht Jupyter die C ++ Fehler-Stdout-Meldungen nicht, die die Geräteplatzierungsprotokollierung verwendet. Hier ist ein längerer Thread zu dem Problem:
Es gibt keine einfache Problemumgehung, die ich kenne, außer das Skript außerhalb von Jupyter auszuführen.
Tags und Links tensorflow