Closed pavo-tusker closed 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 ?
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
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 ?