Installation Guide - RAPIDS Docs (original) (raw)

RAPIDS has several methods for installation, depending on the preferred environment and version.New Users should review the system and environment prerequisites.

Install RAPIDS with Release Selector

System Requirements

Environment Setup

Next Steps

Install RAPIDS

Use the selector tool below to select your preferred method, packages, and environment to install RAPIDS. Certain combinations may not be possible and are dimmed automatically.

Installation Troubleshooting

Conda Issues

A conda create error occurs:
To resolve this error please follow one of these steps:

A __cuda constraint conflict occurs:
You may see something like:

LibMambaUnsatisfiableError: Encountered problems while solving:
 - package cuda-version-12.0-hffde075_0 has constraint __cuda >=12 conflicting with __cuda-11.4-0

This means the CUDA driver currently installed on your machine (e.g. __cuda: 11.4.0) is incompatible with the cuda-version (12.0) you are trying to install. You will have to ensure the CUDA driver on your machine supports the CUDA version you are trying to install with conda.

If conda has incorrectly identified the CUDA driver, you can override by setting the CONDA_OVERRIDE_CUDA environment variable.

Docker Issues

RAPIDS 23.08 brought significant Docker changes.
To learn more about these changes, please see the RAPIDS Container README. Some key notes below:

pip Issues

pip installations require using the matching wheel to the system’s installed CUDA toolkit. For CUDA 11 toolkits, install the -cu11 wheels, and for CUDA 12 toolkits install the -cu12 wheels. If your installation has a CUDA 12 driver but a CUDA 11 toolkit, use the -cu11 wheels.
Infiniband is not supported yet.
These packages are not compatible with Tensorflow pip packages. Please use the NGC containers or conda packages instead.
If you experience a “Failed to import CuPy” error, please uninstall any existing versions of cupy and install cupy-cuda11x. For example:

pip uninstall cupy-cuda115; pip install cupy-cuda11x

The following error message indicates a problem with your environment:

ERROR: Could not find a version that satisfies the requirement cudf-cu12 (from versions: 0.0.1, 25.04)
ERROR: No matching distribution found for cudf-cu12

Check the suggestions below for possible resolutions:

Dask / Jupyter / Tornado 6.2 dependency conflicts can occur. Install jupyter-client 7.3.4 if the error below occurs: 

ERROR: pip's dependency resolver does not currently take into account all the packages that are installed. This behavior is the source of the following dependency conflicts.
jupyter-client 7.4.2 requires tornado>=6.2, but you have tornado 6.1 which is incompatible.

cuSpatial installation may yield the error below: 

ERROR: GDAL >= 3.2 is required for fiona. Please upgrade GDAL.

To resolve, either GDAL needs to be updated, or fiona needs to be pinned to specific versions depending on the installation OS. please see the cuSpatial README to resolve this error.

WSL2 Issues

See the WSL2 setup troubleshooting section.

System Requirements

OS / GPU Driver / CUDA Versions

All provisioned systems need to be RAPIDS capable. Here’s what is required:

GPU: NVIDIA Volta™ or higher with compute capability 7.0+

OS:

CUDA & NVIDIA Drivers: One of the following supported versions:

Note: RAPIDS is tested with and officially supports the versions listed above. Newer CUDA and driver versions may also work with RAPIDS. See CUDA compatibility for details.

CUDA Support

Docker and Conda

pip

System Recommendations

Aside from the system requirements, other considerations for best performance include:

Cloud Instance GPUs

If you do not have access to GPU hardware, there are several cloud service providers (CSP) that are RAPIDS enabled. Learn how to deploy RAPIDS on AWS, Azure, GCP, and IBM cloud on our Cloud Deployment Page.

Several services also offer free and limited trials with GPU resources:

Environment Setup

For most installations, you will need a Conda or Docker environments installed for RAPIDS. Note, these examples are structured for installing on Ubuntu. Please modify appropriately for Rocky Linux. Windows 11 has a WSL2 specific install.

Conda

RAPIDS can be used with any conda distribution.

Below is an installation guide using miniforge.

1. Download and Run Install Script. Copy the command below to download and run the miniforge install script:

curl -L -O "https://github.com/conda-forge/miniforge/releases/latest/download/Miniforge3-$(uname)-$(uname -m).sh"
bash Miniforge3-$(uname)-$(uname -m).sh

2. Customize Conda and Run the Install. Use the terminal window to finish installation. Note, we recommend enabling conda-init.

3. Start Conda. Open a new terminal window, which should now show Conda initialized.

4. Check Conda Configuration. Installing RAPIDS may require you to use channel_priority: flexible.

If you are installing RAPIDS with CUDA 12 or greater, then you can use either strict or flexible channel priority.

If you are install RAPIDS with CUDA 11, then you must set channel_priority: flexible.

You can check this and change it, if required, by doing:

conda config --show channel_priority
conda config --set channel_priority flexible

Docker

RAPIDS requires both Docker CE v19.03+ and nvidia-container-toolkit installed.

1. Download and Install. Copy command below to download and install the latest Docker CE Edition:

curl https://get.docker.com | sh

2. Install Latest NVIDIA Docker. Select the appropriate supported distribution:

curl -s -L https://nvidia.github.io/nvidia-container-runtime/gpgkey | sudo apt-key add - distribution=$(. /etc/os-release;echo <span class="katex"><span class="katex-mathml"><math xmlns="http://www.w3.org/1998/Math/MathML"><semantics><mrow><mi>I</mi><mi>D</mi></mrow><annotation encoding="application/x-tex">ID</annotation></semantics></math></span><span class="katex-html" aria-hidden="true"><span class="base"><span class="strut" style="height:0.6833em;"></span><span class="mord mathnormal" style="margin-right:0.07847em;">I</span><span class="mord mathnormal" style="margin-right:0.02778em;">D</span></span></span></span>VERSION_ID)
curl -s -L https://nvidia.github.io/nvidia-container-runtime/$distribution/nvidia-container-runtime.list | sudo tee /etc/apt/sources.list.d/nvidia-container-runtime.list
sudo apt-get update
sudo apt-get install nvidia-container-runtime

3. Start Docker. In new terminal window run:

sudo service docker stop
sudo service docker start

4a. Test NVIDIA Docker. In a terminal window run:

docker run --gpus all nvcr.io/nvidia/k8s/cuda-sample:nbody nbody -gpu -benchmark

4b. Legacy Docker Users. Docker CE v18 & nvidia-docker2 users will need to replace the following for compatibility:docker run --gpus all with docker run --runtime=nvidia

JupyterLab.

The command provided from the selector for the notebooks Docker image will run JupyterLab on your host machine at port: 8888.

Running Multi-Node / Multi-GPU (MNMG) Environment. To start the container in an MNMG environment:

docker run -t -d --gpus all --shm-size=1g --ulimit memlock=-1 --ulimit stack= 67108864 -v $PWD:/ws <container label>

The standard docker command may be sufficient, but the additional arguments ensures more stability. See the NCCL docs and UCX docs for more details on MNMG usage.

Custom Datasets. See the RAPIDS Container README for more information about using custom datasets. Docker Hub and NVIDIA GPU Cloud host RAPIDS containers with a full list of available tags.

pip

RAPIDS pip packages are available for CUDA 11 and CUDA 12 on the NVIDIA Python Package Index.

pip Additional Prerequisites

The CUDA toolkit version on your system must match the pip CUDA version you install (-cu11 or -cu12).
glibc version: x86_64 wheels require glibc >= 2.17.
glibc version: ARM architecture (aarch64) wheels require glibc >= 2.32 (only ARM Server Base System Architecture is supported).

SDK Manager (Ubuntu Only)

NVIDIA SDK Manager gives a users a Graphical User Interface (GUI) option to install RAPIDS. It also attempts to fix any environment issues before installing RAPIDS or updating RAPIDS, making it ideal for new Linux users.

  1. Download SDK Manager’s Ubuntu version from their website (requires sign up or login to NVIDIA’s Developer community). Do not install yet. It is assumed that your home directory’s Downloads folder is where the .deb file will be stored. If not, please move sdkmanager_[version]-[build#]_amd64.deb file to your current Download folder.
  2. Install and run SDK Manager using the installation guide here. For Ubuntu, use the following commands:
sudo apt install ./sdkmanager_[version]-[build#]_amd64.deb  
sdkmanager
  1. Sign in when asked, and follow SDK Manager’s RAPIDS installation instructions.

Windows WSL2

Windows users can now tap into GPU accelerated data science on their local machines using RAPIDS on Windows Subsystem for Linux 2. WSL2 is a Windows feature that enables users to run native Linux command line tools directly on Windows. Using this feature does not require a dual boot environment, removing complexity and saving you time.

WSL2 Additional Prerequisites

OS: Windows 11 with a WSL2 installation of Ubuntu (minimum version 20.04).
WSL Version: WSL2 (WSL1 not supported).
GPU: GPUs with Compute Capability 7.0 or higher (16GB+ GPU RAM is recommended).

Limitations

Only single GPU is supported.
GPU Direct Storage is not supported.

Troubleshooting

When installing with Conda, if an http 000 connection error occurs when accessing the repository data, run wsl --shutdown and then restart the WSL instance.

When installing with Conda or pip, if an WSL2 Jitify fatal error: libcuda.so: cannot open shared object file error occurs, follow suggestions in this WSL issue to resolve.

When installing with Docker Desktop, if the container pull command is successful, but the run command hangs indefinitely, ensure you’re on Docker Desktop >= 4.18.

WSL2 SDK Manager Install

NVIDIA’s SDK Manager gives Windows users a Graphical User Interface (GUI) option to install RAPIDS. It also attempts to fix any environment issues before installing RAPIDS or updating RAPIDS, making it ideal for new WSL users.

  1. Install the latest NVIDIA Drivers on the Windows host.
  2. Download SDK Manager’s Ubuntu version from their website (requires sign up or login to NVIDIA’s Developer community). Do not install yet. The rest of the instructions assume that your home directory’s Downloads folder is where the .deb file will be stored. If this is not the case, please change the directory, as needed, for your system.
  3. Install or update WSL2 and the Ubuntu 22.04 or Ubuntu 24.04 package using Microsoft’s instructions. To install Ubuntu 24.04 from the command line, use this command:
wsl --install -d Ubuntu-24.04

This will install and start Ubuntu in your Windows host system using WSL2. Make your sudo password memorable as you will need it in the next two steps. 4. Install and run SDK Manager inside Ubuntu by pasting this into your command line. This command will navigate to your Windows users’s Downloads folder, from your WSL2 instance, and install the latest SDK Manager .deb file that you had downloaded. You will have to enter the sudo password you created when you installed Ubuntu. 

sudo apt update && sudo apt install wslu -y  
cd "$(wslpath -au "$(cmd.exe /c 'echo %USERPROFILE%' | tr -d '\r')")/Downloads"  
sudo apt install "$(ls -t ./sdkmanager_*_amd64.deb | head -n 1)" -y  
sdkmanager
  1. Sign in when asked, and follow SDK Manager’s RAPIDS installation instructions here.

WSL2 Conda Install

  1. Install WSL2 and the Ubuntu distribution using Microsoft’s instructions.
  2. Install the latest NVIDIA Drivers on the Windows host.
  3. Log in to the WSL2 Linux instance.
  4. Install Conda in the WSL2 Linux Instance using our Conda instructions.
  5. Install RAPIDS via Conda, using the RAPIDS Release Selector.
  6. Run this code to check that the RAPIDS installation is working:
import cudf  
print(cudf.Series([1, 2, 3]))

WSL2 Docker Desktop Install

  1. Install WSL2 and the Ubuntu distribution using Microsoft’s instructions.
  2. Install the latest NVIDIA Drivers on the Windows host.
  3. Install latest Docker Desktop for Windows
  4. Log in to the WSL2 Linux instance.
  5. Generate and run the RAPIDS docker command based on your desired configuration using the RAPIDS Release Selector.
  6. Inside the Docker instance, run this code to check that the RAPIDS installation is working:
import cudf  
print(cudf.Series([1, 2, 3]))

WSL2 pip Install

  1. Install WSL2 and the Ubuntu distribution using Microsoft’s instructions.
  2. Install the latest NVIDIA Drivers on the Windows host.
  3. Log in to the WSL2 Linux instance.
  4. Follow this helpful developer guide and then install the WSL-specific CUDA 11 or CUDA 12 Toolkit without drivers into the WSL2 instance.
    • The installed CUDA Toolkit version must match the pip wheel version (-cu11 or -cu12)
    • Any CUDA 12 CTK will work with RAPIDS -cu12 pip packages
  5. Install RAPIDS pip packages on the WSL2 Linux Instance using the release selector commands.
  6. Run this code to check that the RAPIDS installation is working:
import cudf  
print(cudf.Series([1, 2, 3]))

Build from Source

To build from source, check each RAPIDS GitHub README, such as the cuDF’s source environment set up and build instructions. Further links are provided in the selector tool. If additional help is needed reach out on our Slack Channel.

Next Steps

After installing the RAPIDS libraries, the best place to get started is our User Guide. Our RAPIDS.ai home page also provides a great deal of information, as does our Blog Page and the NVIDIA Developer Blog. We are also always available on our RAPIDS GoAi Slack Channel.