Submitted by Realistic-Bed2658 t3_zzhg84 in MachineLearning
AerysSk t1_j2cgmo4 wrote
Reply to comment by Realistic-Bed2658 in [D] GPU-enabled scikit-learn by Realistic-Bed2658
If you are looking for a GPU version of scikit-learn, I think Nvidia is making one, and they call it cuml. Note that, not all algorithms are implemented, and there will also be some missing functions as well.
However, a note about Apple and AMD GPU thing: they are on the rise, but not until a few years later that they will become usable. My lab has only Nvidia GPUs but we already have a lot of headache dealing with Nvidia drivers and libraries. At least for a few years, we do not see any plan switching to AMD or Apple.
JocialSusticeWarrior t1_j2ed75n wrote
unfortunate name "cuml"
Realistic-Bed2658 OP t1_j2cld4m wrote
Totally understandable. I only use nvidia at work too.
Thanks for the info about the nvidia package!
Viewing a single comment thread. View all comments