Wie verwenden Sie multi-CPU-Kerne zum Zug NNs mit caffe und OpenBLAS

Ich Lerne deep learning vor kurzem und mein Freund empfahl mir caffe. Nach der Installation es mit OpenBLAS, ich folgte dem tutorial, MNIST Aufgabe in der doc. Aber später fand ich, war es super langsam ist und nur einen CPU-Kern arbeiten.

Das problem ist, dass die Server in meinem Labor nicht über die GPU, also muss ich verwenden CPUs statt.

Googelte ich dieses und bekam einige Seite wie diese . Ich habe versucht, zu export OPENBLAS_NUM_THREADS=8 und export OMP_NUM_THREADS=8. Aber caffe noch verwendet einen Kern.

Wie kann ich machen, caffe verwenden Sie multi-CPUs?

Vielen Dank.

  • Haben Sie openblas zur Benutzung von threads?
  • Ich habe gerade make und make install. Ich fand eine page, aber es sagt nichts über Gebäude es zur Benutzung von threads. Wie kann ich es bauen, verwenden threads?
  • Lesen Sie die Dokumentation. Es ist ziemlich klar.
  • Ich muss sagen, ich habe nicht überall erwähnt, kompilieren OpenBLAS mit jedem Parameter in Bezug auf threads.
  • USE_OPENMP=1 ist bereits in github.com/xianyi/OpenBLAS/blob/develop/README.md. Das ist, wie ich immer bauen für Gewinde-Nutzung.
  • Leider caffe noch eine CPU verwendet, nachdem ich rebuild OpenBLAS mit USE_OPENMP=1 und dann neu erstellen caffe. 🙁
  • Was ist deine system-config?
  • Centos 6.5, 24-core-CPU. Meinst du diese?
  • Wie Sie wissen, Caffe ist nur mit einem Kern? Die zeitliche Auflösung top möglicherweise nicht genug, um zu fangen dgemm in Aktion. Haben Sie gprof um zu sehen, ob eine Erhöhung OMP_NUM_THREADS wirkt sich auf die Wand der Zeit?
  • Ich deinstallieren und neu installieren das ganze und es funktioniert. Aber auch wenn ich alle nutzen die cpu, es ist immer noch suuuper langsam. Nun, die gute Nachricht ist, dass mein Chef kaufte einen TITAN-X für das Labor. LOL
  • Es ist eine OpenMP-version von Caffe auf Github, die im Wettbewerb mit der GPU-Hafen für einige workloads. Sie könnten versuchen, es zu finden. Threading-GEMM ist nicht immer der beste Weg, um DNN schneller gehen...
  • bitte post den link zu OpenMP caffe Gabel.
  • Siehe github.com/intelcaffe.
  • was ist das Besondere an dieser Gabel? Ich kann nicht sehen, alle Verweise auf openmp.
  • Sorry, Sie sind in den Prozess der überarbeitung der OpenMP-Zeug. Es ist github.com/intelcaffe/caffe-old/tree/openmp, aber ich vermute, Sie sehen wollen, z.B. github.com/intelcaffe/caffe/commits/openmp-conv-relu.

InformationsquelleAutor magic282 | 2015-05-12
Schreibe einen Kommentar