R: Maschinelles Lernen auf der GPU

15

Gibt es maschinelle Lernpakete für R, die die GPU nutzen können, um die Trainingsgeschwindigkeit zu verbessern (so etwas wie Theano aus der Python-Welt)?

Ich sehe, dass es ein Paket namens gputools gibt, das die Ausführung von Code auf der GPU ermöglicht, aber ich suche nach einer vollständigeren Bibliothek für maschinelles Lernen.

machine-learning r gpu Simon
quelle

Antworten:

15

Für ein vollständiges Paket für maschinelles Lernen auf GPUs gibt es kein solches Paket. Es gibt jedoch tatsächlich eine Handvoll R-Pakete, die GPUs verwenden können. Sie können diese Pakete auf der Seite CRAN High Performance Computing sehen . Beachten Sie, dass für die meisten dieser Pakete eine NVIDIA-Karte erforderlich ist. Von den verfügbaren Paketen gibt es drei Pakete, die Sie höchstwahrscheinlich verwenden würden, es sei denn, Sie haben einen Sonderfall.

  1. gputools - wenn Sie an Entfernungsberechnungen interessiert sind (nur NVIDIA).
  2. gmatrix - allgemeine numerische Berechnungen (nur NVIDIA).
  3. gpuR - allgemeine numerische Berechnungen (jede GPU über OpenCL). *

* HINWEIS - Auf die Gefahr der Eigenwerbung bin ich der Autor des gpuR-Pakets.

Sie können wahrscheinlich die beiden letztgenannten Pakete verwenden, um vorhandene Algorithmen für maschinelles Lernen zu reproduzieren. Ich verwende mein gpuR-Paket tatsächlich, um ein GPU-beschleunigtes Neuralnet-Paket zu erstellen, aber dies ist in Bearbeitung.

Zusammenfassend ist festzuhalten, dass die grundlegenden Ressourcen in R verfügbar sind. Wenn Sie jedoch in naher Zukunft etwas benötigen, müssen Sie andere Ressourcen / Ansätze untersuchen, auf die @YCR hingewiesen hat.

cdeterman
quelle
irgendwelche Fortschritte bei den neuronalen Netzfähigkeiten von gpuR? Ein NEWSLink könnte hilfreich sein, wenn nicht :)
MichaelChirico
Die gpuR NEWS sind da . Die neuronalen Netzfunktionen sind noch in Bearbeitung. Es gibt mehrere Pakete, die daran beteiligt sind, da jedes einzeln verwendet werden kann (Lazytensor, gpuRNN & prometheus). Du kannst meinem Github folgen, wenn du neugierig bist. Ich bin nur ein Entwickler, der in meiner Freizeit arbeitet und ich muss meine anderen Pakete weiterentwickeln und warten.
Cdeterman
2

Dies ist wirklich ein Wrapper über Tensorflow, Caffe, Mxnet, aber möglicherweise nützlich für Sie.

https://www.h2o.ai/deep-water/

HoraceT
quelle
1

Eine gute Bibliothek für maschinelles Lernen mit GPUs ist mxnet . Das Paket ist jedoch größtenteils Deep Learning. Wenn Sie also nach bestimmten Algorithmen für maschinelles Lernen suchen, finden Sie sie möglicherweise dort nicht. Sie verfügen jedoch über eine Reihe guter Deep Learning-Algorithmen.

Hoaphumanoid
quelle