cuml icon indicating copy to clipboard operation
cuml copied to clipboard

[FEA] Simplify process to train cuml KMeans on GPU and save the model and later load on a CPU machine for inference

Open user06039 opened this issue 3 years ago • 9 comments

I am trying to use KMeans in CUML for fitting the data, but for inference/prediction I want to do it on CPU? Is it possible somehow? I really need a way to predict on CPU. Please help

EDIT:

I feel like it's useful feature for the community, since training and tuning is more resource taking process using GPU make sense but for inference I feel like a CPU machine should do a decent job in production.

user06039 avatar Mar 16 '21 20:03 user06039