cuML - RAPIDS Machine Learning Library
Branch: master
Clone or download
Pull request Compare This branch is 181 commits behind branch-0.6.
Fetching latest commit…
Cannot retrieve the latest commit at this time.
Type Name Latest commit message Commit time
Failed to load latest commit information.
cuML ENH Update cuML and cmake versions Jan 29, 2019
docs REL v0.5.1 cuML release Feb 5, 2019
img ENH Add RAPIDS images for readme Dec 14, 2018
python FIX Remove unnecessary import numpy Feb 5, 2019
.gitattributes Fix cuML->cuml references in docs and notebooks, start using versione… Nov 4, 2018
.gitignore Merge branch 'branch-0.5' into consolidate-ml-prims Dec 14, 2018
LICENSE Update License Info Oct 9, 2018 Fix cuML->cuml references in docs and notebooks, start using versione… Nov 4, 2018 REL Update conda install command for v0.5 Feb 1, 2019
readthedocs.yml ENH Match conda dirs with cuDF Jan 28, 2019
requirements.txt BLD Add for pip builds Jan 22, 2019

 cuML - RAPIDS Machine Learning Algorithms

Machine learning is a fundamental capability of RAPIDS. cuML is a suite of libraries that implements a machine learning algorithms within the RAPIDS data science ecosystem. cuML enables data scientists, researchers, and software engineers to run traditional ML tasks on GPUs without going into the details of CUDA programming.

NOTE: For the latest stable ensure you are on the master branch.

The cuML repository contains:

  1. python: Python based GPU Dataframe (GDF) machine learning package that takes cuDF dataframes as input. cuML connects the data to C++/CUDA based cuML and ml-prims libraries without ever leaving GPU memory.

  2. cuML: C++/CUDA machine learning algorithms. This library currently includes the following six algorithms; a) Single GPU Truncated Singular Value Decomposition (tSVD), b) Single GPU Principal Component Analysis (PCA), c) Single GPU Density-based Spatial Clustering of Applications with Noise (DBSCAN), d) Single GPU Kalman Filtering, e) Multi-GPU K-Means Clustering, f) Multi-GPU K-Nearest Neighbors (Uses Faiss).

  3. ml-prims: Low level machine learning primitives used in cuML. ml-prims is comprised of the following components; a) Linear Algebra, b) Statistics, c) Basic Matrix Operations, d) Distance Functions, e) Random Number Generation.

Available Algorithms:

  • Truncated Singular Value Decomposition (tSVD),

  • Principal Component Analysis (PCA),

  • Density-based spatial clustering of applications with noise (DBSCAN),

  • K-Means Clustering,

  • K-Nearest Neighbors (Requires Faiss installation to use),

  • Linear Regression (Ordinary Least Squares),

  • Ridge Regression.

  • Kalman Filter.

Upcoming algorithms:

  • More Kalman Filter versions,

  • Lasso,

  • Elastic-Net,

  • Logistic Regression,

  • UMAP

More ML algorithms in cuML and more ML primitives in ml-prims are being added currently. Example notebooks are provided in the python folder to test the functionality and performance. Goals for future versions include more algorithms and multi-gpu versions of the algorithms and primitives.

The installation option provided currently consists on building from source. Upcoming versions will add pip and conda options, along docker containers. They will be available in the coming weeks.



cuML can be installed using the rapidsai conda channel:

conda install -c nvidia -c rapidsai -c conda-forge -c pytorch -c defaults cuml=0.5


cuML can also be installed using pip. Select the package based on your version of CUDA:

# cuda 9.2
pip install cuml-cuda92

# cuda 10.0
pip install cuml-cuda100

You also need to ensure libomp and libopenblas are installed:

apt install libopenblas-base libomp-dev

Note: There is no faiss-gpu package installable by pip, so the KNN algorithm will not work unless you install Faiss manually or via conda (see below).

Dependencies for Installing/Building from Source:

To install cuML from source, ensure the dependencies are met:

  1. cuDF (>=0.5.0)
  2. zlib Provided by zlib1g-dev in Ubuntu 16.04
  3. cmake (>= 3.12.4)
  4. CUDA (>= 9.2)
  5. Cython (>= 0.29)
  6. gcc (>=5.4.0)
  7. faiss-gpu (>=1.4.0) - faiss-gpu is required to run the KNN algorithm.
    • For using KNN with CUDA 9.2 or CUDA 10.0, faiss-gpu can be installed using conda:
# cuda 9.2
conda install -c pytorch faiss-gpu cuda92

# cuda 10.0
conda install -c pytorch faiss-gpu cuda100

Installing from Source:

Once dependencies are present, follow the steps below:

  1. Clone the repository.
$ git clone --recurse-submodules
  1. Build and install libcuml (the C++/CUDA library containing the cuML algorithms), starting from the repository root folder:
$ cd cuML
$ mkdir build
$ cd build
$ cmake ..

Note: if using a conda environment (recommended currently), then cmake can be configured appropriately via:

  1. Build libcuml:
$ make -j
$ make install

To run tests (optional):

$ ./ml_test

If you want a list of the available tests:

$ ./ml_test --gtest_list_tests
  1. Build the cuml python package:
$ cd ../../python
$ python build_ext --inplace

To run Python tests (optional):

$ py.test -v

If you want a list of the available tests:

$ py.test cuML/test --collect-only
  1. Finally, install the Python package to your Python path:
$ python install

Python Notebooks

Demo notebooks for the cuML Python algorithms can be found in the rapidsai/notebooks repository on Github.


The external folders contains submodules that this project in-turn depends on. Appropriate location flags will be automatically populated in the main CMakeLists.txt file for these.

Current external submodules are:


Please use issues and pull requests to report bugs and add functionality.


Find out more details on the RAPIDS site

Open GPU Data Science

The RAPIDS suite of open source software libraries aim to enable execution of end-to-end data science and analytics pipelines entirely on GPUs. It relies on NVIDIA® CUDA® primitives for low-level compute optimization, but exposing that GPU parallelism and high-bandwidth memory speed through user-friendly Python interfaces.