TensorFlow-Operationen mit GPU-Unterstützung

Question

Gibt es einen Weg (oder vielleicht eine Liste?), Um alle Tensorflow-Operationen zu ermitteln, die GPU-Unterstützung bieten? Im Moment ist es ein Versuch und Irrtum für mich - ich versuche, eine Gruppe von Operationen auf GPU zu platzieren. Wenn es funktioniert, süß. Wenn nicht, versuchen Sie es woanders.

Das ist das einzige, was relevant (aber nicht hilfreich) die ich bisher gefunden habe: https://github.com/tensorflow/tensorflow/issues/2502

Answer 1

Leider scheint es, gibt es keine integrierte Möglichkeit, eine genaue Liste oder sogar überprüfen Sie dies für eine bestimmte op zu erhalten. Wie in der obigen Ausgabe erwähnt, erfolgt die Verteilung im nativen C++ - Code: Eine bestimmte Operation kann GPU zugewiesen werden, wenn ein entsprechender Kernel unter DEVICE_GPU registriert wurde.

Ich denke, der einfachste Weg für Sie ist grep "REGISTER_KERNEL_BUILDER" -r tensorflow die Tensorflow-Quellbasis, um eine Liste der passenden Operationen, die etwas aussehen wird like this.

Aber denken Sie daran, dass selbst mit REGISTER_KERNEL_BUILDER Spezifikation, gibt es keine Garantie, dass ein Op auf einer GPU durchgeführt wird. Beispielsweise wird 32-Bit-Int Add auf CPU unabhängig vom vorhandenen Kernel zugewiesen.