Ich installiere Caffe auf einem virtuellen Ubuntu 14.04 Server mit CUDA installiert (ohne Treiber) mit Ссылка als Inspiration. Während des Installationsvorgangs habe ich das MakeFile so bearbeitet, dass es "CPU_ONLY := 1" enthält, bevor es erstellt wird. Es scheint jedoch, dass Caffe immer noch versucht, die GPU zu nutzen. Wenn ich versuche, ein Testbeispiel auszuführen, erhalte ich den folgenden Fehler:
Wie kann ich das beheben und vollständig auf der CPU laufen?
Ich werde einige Worte zu Mailerdaimons Antwort hinzufügen.
Ich folgte dem Installationsleitfaden ( Ссылка ), um Caffe in meinem virtuellen Vagrant einzurichten Maschine. Zu Ihrer Information, virtuelle Maschinen unterstützen keine GPU-Beschleunigung. Zurück zu dem Punkt, nachdem ich "CPU / GPU-Schalter in Beispielskripten" ( Ссылка ) korrigiert und hinzugefügt habe "- -print_results --labels_file 'Optionen ( Ссылка ) zu' python / classify.py ' , dieser Befehl './python/classify.py ./examples/images/cat.jpg foo --print_results' löst immer noch den folgenden Fehler aus:
%Vor%Dann lasse ich die Form von 'mean' (was 3 * 256 * 256 ist) und 'input' (3 * 227 * 227). Offensichtlich sind diese beiden Formen inkompatibel. Aber alte Versionen von 'set_mean ()' werfen den Fehler NICHT, also tauche ich in den Python-Code und finde heraus, dass die alte 'set_mean ()' -Funktion wie folgt aussieht (python / caffe / pycaffe.py, Zeile 195-202, Ссылка ):
%Vor%Aber im letzten Caffe kapseln die Mitwirkenden 'set_mean ()' und andere Transformationsfunktionen in die Klasse ein 'Transformator'. Die neue Funktion 'set_mean ()' sieht so aus (python / caffe / io.py, Zeile 253-254, Ссылка ):
%Vor% Jesus, wie könnten diese beiden die gleiche Funktion sein? Also ändere ich die neue 'set_mean ()', kommentiere den fehlererhöhenden Satz aus und füge einen Shape-Sizing-Vorgang wie im alten 'set_mean ()' hinzu. %Vor%Voila, Problem gelöst.
%Vor%Derzeit gibt es einige Probleme, die auf eine Vielzahl von Schnittstellenänderungen zurückzuführen sind, die von den Caffe-Entwicklern eingeführt wurden. Der Python Wrapper wurde noch nicht mit diesen Änderungen aktualisiert.
Sehen Sie diese PR, die das Problem behebt: Ссылка
Derselbe Fehler wurde in einer anderen Google-Gruppe behoben: -
Sie müssen das nur ändern:
%Vor%dazu:
%Vor%Google Group: Mittlere Form, die mit der Eingabeform nicht kompatibel ist