Skip to content

RAPIDS

Suite of libraries for data analytics and machine learning on GPUs.

Available

Rapids is available on Puhti and Mahti. Some older versions are available on Puhti only. Currently supported RAPIDS versions:

  • 21.12 using Singularity, based on RAPIDS official Docker images: 21.12
  • 0.16 using Singularity: 0.16-sng (Puhti only)
  • 0.15 using Singularity: 0.15-sng (Puhti only)
  • 0.14 using Singularity: 0.14-sng (Puhti only)
  • 0.11 (Puhti only)

Contains the RAPIDS suite (including cuDF, cuML, cuGraph, and XGBoost) for Python with GPU support via CUDA.

Modules ending with -sng are based on NVIDIA's optimized container images from NGC.

If you find that some package is missing, you can often install it yourself with pip install --user. If you think that some important RAPIDS-related package should be included in the module provided by CSC, you can send an email to servicedesk@csc.fi.

Some modules are Singularity-based. Wrapper scripts have been provided so that common commands such as python, python3, pip and pip3 should work as normal. For more information, see CSC's general instructions on how to run Singularity containers.

License

RAPIDS is licensed under Apache License 2.0

Usage

To use this software, initialize it with:

module load rapids

to access the default version.

This will show all available versions:

module avail rapids

To check the exact packages and versions included in the loaded module you can run:

list-packages

Note

Note that login nodes are not intended for heavy computing, please use slurm batch jobs instead. See our instructions on how to use the batch job system.

Local storage

The GPU nodes have fast local storage which is useful for IO-intensive applications. See our general instructions on how to take the fast local storage into use.

More information

Last edited Wed Jan 26 2022