R: aprendizaje automático en GPU


15

¿Hay algún paquete de aprendizaje automático para R que pueda hacer uso de la GPU para mejorar la velocidad de entrenamiento (algo así como theano del mundo python)?

Veo que hay un paquete llamado gputools que permite la ejecución de código en la gpu, pero estoy buscando una biblioteca más completa para el aprendizaje automático.

Respuestas:


15

En cuanto a un paquete completo de aprendizaje automático en GPU, no existe tal paquete. Sin embargo, en realidad hay un puñado de paquetes R que pueden usar GPU. Puede ver estos paquetes en la página de Computación de alto rendimiento de CRAN . Debe tener en cuenta que la mayoría de estos paquetes requieren que tenga una tarjeta NVIDIA. De los paquetes disponibles, hay tres paquetes que probablemente utilizaría a menos que tenga un caso especial.

  1. gputools : si está interesado en cálculos de distancia (solo NVIDIA).
  2. gmatrix : cálculos numéricos generales (solo NVIDIA).
  3. gpuR : cálculos numéricos generales (cualquier GPU a través de OpenCL). *

* NOTA - A riesgo de autopromoción, soy el autor del paquete gpuR.

Es probable que pueda usar los dos últimos paquetes para reproducir los algoritmos de aprendizaje automático existentes. De hecho, estoy usando mi paquete gpuR para crear un paquete de red neuronal acelerado por GPU, pero esto está en progreso.

En resumen, si está determinado, los recursos básicos están disponibles en R. Pero si necesita algo en el futuro inmediato, necesitará explorar otros recursos / enfoques como lo señala @YCR.


¿Hay algún progreso en las capacidades de la red neuronal de gpuR? Un NEWSenlace podría ser útil si no :)
MichaelChirico

Las NOTICIAS gpuR están aquí . Las capacidades de la red neuronal todavía están en progreso. Hay varios paquetes que estarán involucrados con esto, ya que cada uno se puede usar de forma independiente (lazytensor, gpuRNN y prometheus). Puedes seguir mi github si tienes curiosidad. Solo soy un desarrollador que trabaja en mis momentos libres y necesito continuar desarrollando y manteniendo mis otros paquetes.
cdeterman



1

Una buena biblioteca para el aprendizaje automático con GPU es mxnet . Sin embargo, el paquete es principalmente de aprendizaje profundo, por lo que si está buscando algoritmos específicos de aprendizaje automático, es posible que no los encuentre allí. Sin embargo, tienen un buen conjunto de algoritmos de aprendizaje profundo.

Al usar nuestro sitio, usted reconoce que ha leído y comprende nuestra Política de Cookies y Política de Privacidad.
Licensed under cc by-sa 3.0 with attribution required.