R: apprendimento automatico su GPU


15

Esistono pacchetti di apprendimento automatico per R che possono utilizzare la GPU per migliorare la velocità di allenamento (qualcosa come theano dal mondo Python)?

Vedo che esiste un pacchetto chiamato gputools che consente l'esecuzione di codice sulla gpu, ma sto cercando una libreria più completa per l'apprendimento automatico.

Risposte:


15

Per quanto riguarda un pacchetto di apprendimento automatico completo sulle GPU, non esiste un pacchetto del genere. Tuttavia, ci sono in realtà una manciata di pacchetti R che possono usare le GPU. Puoi vedere questi pacchetti nella pagina CRAN High Performance Computing . Si noti che la maggior parte di questi pacchetti richiede la presenza di una scheda NVIDIA. Dei pacchetti disponibili, ci sono tre pacchetti che molto probabilmente useresti a meno che tu non abbia un caso speciale.

  1. gputools - se interessati ai calcoli a distanza (solo NVIDIA).
  2. gmatrix - calcoli numerici generali (solo NVIDIA).
  3. gpuR - calcoli numerici generali (qualsiasi GPU tramite OpenCL). *

* NOTA - A rischio di autopromozione sono l'autore del pacchetto gpuR.

Probabilmente puoi usare questi ultimi due pacchetti per riprodurre algoritmi di machine learning esistenti. In realtà sto usando il mio pacchetto gpuR per creare un pacchetto neuralnet accelerato GPU ma questo è in corso.

Quindi, in sintesi, se sei determinato, le risorse di base sono disponibili in R. Ma se hai bisogno di qualcosa nell'immediato futuro, dovrai esplorare altre risorse / approcci come sottolineato da @YCR.


qualche progresso sulle capacità della rete neurale di gpuR? Un NEWSlink potrebbe essere utile in caso contrario :)
MichaelChirico l'

La NEWS gpuR è qui . Le capacità della rete neurale sono ancora in corso. Ci sono più pacchetti che saranno coinvolti in questo poiché ognuno può essere usato indipendentemente (lazytensor, gpuRNN e prometheus). Puoi seguire il mio github se sei curioso. Sono solo uno sviluppatore che lavora nei miei momenti liberi e devo continuare a sviluppare e mantenere i miei altri pacchetti.
cdeterman,



1

Una buona libreria per l'apprendimento automatico con GPU è mxnet . Il pacchetto è principalmente di deep learning, quindi se stai cercando specifici algoritmi di machine learning potresti non trovarli lì. Tuttavia hanno una buona serie di algoritmi di deep learning.

Utilizzando il nostro sito, riconosci di aver letto e compreso le nostre Informativa sui cookie e Informativa sulla privacy.
Licensed under cc by-sa 3.0 with attribution required.