R: pembelajaran mesin pada GPU

15

Apakah ada paket pembelajaran mesin untuk R yang dapat memanfaatkan GPU untuk meningkatkan kecepatan pelatihan (seperti theano dari dunia python)?

Saya melihat bahwa ada paket yang disebut gputools yang memungkinkan eksekusi kode pada GPU, tetapi saya sedang mencari perpustakaan yang lebih lengkap untuk pembelajaran mesin.

Simon
sumber

Jawaban:

15

Adapun paket pembelajaran mesin lengkap pada GPU, tidak ada paket seperti itu. Namun, sebenarnya ada beberapa paket R yang dapat menggunakan GPU. Anda dapat melihat paket ini di halaman CRAN High Performance Computing . Anda harus mencatat bahwa sebagian besar paket ini mengharuskan Anda untuk memiliki kartu NVIDIA. Dari paket yang tersedia, ada tiga paket yang kemungkinan besar akan Anda manfaatkan kecuali Anda memiliki kasing khusus.

  1. gputools - jika tertarik pada perhitungan jarak (hanya NVIDIA).
  2. gmatrix - perhitungan numerik umum (hanya NVIDIA).
  3. gpuR - perhitungan numerik umum (GPU apa pun melalui OpenCL). *

* CATATAN - Dengan risiko promosi diri saya adalah pembuat paket gpuR.

Anda mungkin dapat menggunakan dua paket terakhir untuk mereproduksi algoritma pembelajaran mesin yang ada. Saya sebenarnya menggunakan paket gpuR saya untuk membuat paket neuralnet yang dipercepat GPU tetapi ini sedang berlangsung.

Jadi secara ringkas, jika Anda ditentukan, sumber daya dasar tersedia di R. Tetapi jika Anda membutuhkan sesuatu dalam waktu dekat, Anda perlu mengeksplorasi sumber daya / pendekatan lain seperti yang ditunjukkan oleh @YCR.

cdeterman
sumber
ada kemajuan pada kemampuan jaringan saraf gpuR? Sebuah NEWSlink yang mungkin bisa membantu jika tidak :)
MichaelChirico
NEWS gpuR ada di sini . Kemampuan jaring saraf masih dalam proses. Ada beberapa paket yang akan terlibat dengan ini karena masing-masing dapat digunakan secara independen (lazytensor, gpuRNN, & prometheus). Anda dapat mengikuti github saya jika Anda penasaran. Saya hanya satu pengembang yang bekerja di waktu senggang saya dan saya harus terus mengembangkan dan memelihara paket saya yang lain.
cdeterman
2

Ini benar-benar pembungkus untuk tensorflow, caffe, mxnet, tetapi mungkin berguna bagi Anda.

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

horaceT
sumber
1

Perpustakaan yang baik untuk pembelajaran mesin dengan GPU adalah mxnet . Paket ini sebagian besar pembelajaran mendalam, jadi jika Anda mencari algoritma pembelajaran mesin tertentu, Anda mungkin tidak menemukannya di sana. Namun mereka memiliki satu set algoritma pembelajaran yang baik.

hoaphumanoid
sumber