search

NicolasHug / Surprise

A Python scikit for building and analyzing recommender systems
http://surpriselib.com
BSD 3-Clause "New" or "Revised" License
6.34k stars 1.01k forks source link

Rapids.ai #394

Closed pavo-tusker closed 3 years ago

pavo-tusker commented 3 years ago

Hi Surprise Team,

We are using surprise for recommendations and rapids.ai based libraries to leverage GPU.

Is it possible to support Nvidia GPU / Rapids framework ?

NicolasHug commented 3 years ago

Hi,

No, surprise doesn't support GPU. I'd recommend checking out libraries implemented on top of pytorch or tensorflow, like https://www.tensorflow.org/recommenders