What it does
cuML is a library from NVIDIA that runs classic machine-learning methods, such as clustering and classification, on NVIDIA graphics cards (GPUs).
It works like scikit-learn, a popular Python machine-learning library. It can also speed up existing scikit-learn code without changing it.
Its makers say it can be up to 50 times faster on some tasks, depending on the method, data and hardware.
Who it's for
Data scientists who use Python and have access to NVIDIA graphics cards.
Can I use it without coding?
Developers only. A Python library that needs an NVIDIA GPU. New to this? Start here.
Can I run it?
- Cost
- FreeNothing to pay or sign up for.
- Runs on
- Linux
- Computer
- Works best on a powerful computer or one with a graphics card (GPU)
- Privacy
- Runs on your own device: your data stays with you
- API key
- Not needed
Try it
Using Claude Code, Cursor or another AI coding assistant? Paste the prompt and it will walk you through the setup.New to this?
Before you install
- Only download from the official GitHub page. Copies with similar names exist for many popular projects.
- License (Apache-2.0): Permissive: free to use, change and share, including for business.
- More on staying safe
Where people are talking about it
- CUDA 11.0Hacker News · Jul 2020
- Machine Learning in Python: Main developments and technology trends in data science, machine learning, and artificial intelligenceResearch paper · Feb 2020
- CUDA Toolkit Release NotesHacker News · Nov 2019