Add CUDA 13.1 support (#1642)

bdice · web-flow · commit ef98fb2efff2 · 2026-01-05T22:03:07.000Z
Contributes to rapidsai/build-planning#236 - Adds CUDA 13.1 to dependencies.yaml - Adds CUDA 13.1 devcontainers - Updates install documentation if needed Authors: - Bradley Dice (https://github.com/bdice) - Gil Forsyth (https://github.com/gforsyth) Approvers: - Gil Forsyth (https://github.com/gforsyth) - Ben Frederickson (https://github.com/benfred) URL: #1642
diff --git a/.devcontainer/cuda13.1-conda/devcontainer.json b/.devcontainer/cuda13.1-conda/devcontainer.json
@@ -3,15 +3,15 @@
     "context": "${localWorkspaceFolder}/.devcontainer",
     "dockerfile": "${localWorkspaceFolder}/.devcontainer/Dockerfile",
     "args": {
-      "CUDA": "13.0",
+      "CUDA": "13.1",
       "PYTHON_PACKAGE_MANAGER": "conda",
       "BASE": "rapidsai/devcontainers:26.02-cpp-mambaforge"
     }
   },
   "runArgs": [
     "--rm",
     "--name",
-    "${localEnv:USER:anon}-rapids-${localWorkspaceFolderBasename}-26.02-cuda13.0-conda",
+    "${localEnv:USER:anon}-rapids-${localWorkspaceFolderBasename}-26.02-cuda13.1-conda",
     "--ulimit",
     "nofile=500000"
   ],
@@ -22,7 +22,7 @@
   "overrideFeatureInstallOrder": [
     "ghcr.io/rapidsai/devcontainers/features/rapids-build-utils"
   ],
-  "initializeCommand": ["/bin/bash", "-c", "mkdir -m 0755 -p ${localWorkspaceFolder}/../.{aws,cache,config,conda/pkgs,conda/${localWorkspaceFolderBasename}-cuda13.0-envs}"],
+  "initializeCommand": ["/bin/bash", "-c", "mkdir -m 0755 -p ${localWorkspaceFolder}/../.{aws,cache,config,conda/pkgs,conda/${localWorkspaceFolderBasename}-cuda13.1-envs}"],
   "postAttachCommand": ["/bin/bash", "-c", "if [ ${CODESPACES:-false} = 'true' ]; then . devcontainer-utils-post-attach-command; . rapids-post-attach-command; fi"],
   "workspaceFolder": "/home/coder",
   "workspaceMount": "source=${localWorkspaceFolder},target=/home/coder/cuvs,type=bind,consistency=consistent",
@@ -31,7 +31,7 @@
     "source=${localWorkspaceFolder}/../.cache,target=/home/coder/.cache,type=bind,consistency=consistent",
     "source=${localWorkspaceFolder}/../.config,target=/home/coder/.config,type=bind,consistency=consistent",
     "source=${localWorkspaceFolder}/../.conda/pkgs,target=/home/coder/.conda/pkgs,type=bind,consistency=consistent",
-    "source=${localWorkspaceFolder}/../.conda/${localWorkspaceFolderBasename}-cuda13.0-envs,target=/home/coder/.conda/envs,type=bind,consistency=consistent"
+    "source=${localWorkspaceFolder}/../.conda/${localWorkspaceFolderBasename}-cuda13.1-envs,target=/home/coder/.conda/envs,type=bind,consistency=consistent"
   ],
   "customizations": {
     "vscode": {
diff --git a/.devcontainer/cuda13.1-pip/devcontainer.json b/.devcontainer/cuda13.1-pip/devcontainer.json
@@ -3,22 +3,22 @@
     "context": "${localWorkspaceFolder}/.devcontainer",
     "dockerfile": "${localWorkspaceFolder}/.devcontainer/Dockerfile",
     "args": {
-      "CUDA": "13.0",
+      "CUDA": "13.1",
       "PYTHON_PACKAGE_MANAGER": "pip",
-      "BASE": "rapidsai/devcontainers:26.02-cpp-cuda13.0-ucx1.19.0-openmpi5.0.7"
+      "BASE": "rapidsai/devcontainers:26.02-cpp-cuda13.1-ucx1.19.0-openmpi5.0.7"
     }
   },
   "runArgs": [
     "--rm",
     "--name",
-    "${localEnv:USER:anon}-rapids-${localWorkspaceFolderBasename}-26.02-cuda13.0-pip",
+    "${localEnv:USER:anon}-rapids-${localWorkspaceFolderBasename}-26.02-cuda13.1-pip",
     "--ulimit",
     "nofile=500000"
   ],
   "hostRequirements": {"gpu": "optional"},
   "features": {
     "ghcr.io/rapidsai/devcontainers/features/cuda:26.2": {
-      "version": "13.0",
+      "version": "13.1",
       "installcuBLAS": true,
       "installcuSOLVER": true,
       "installcuRAND": true,
@@ -31,15 +31,15 @@
     "ghcr.io/rapidsai/devcontainers/features/cuda",
     "ghcr.io/rapidsai/devcontainers/features/rapids-build-utils"
   ],
-  "initializeCommand": ["/bin/bash", "-c", "mkdir -m 0755 -p ${localWorkspaceFolder}/../.{aws,cache,config/pip,local/share/${localWorkspaceFolderBasename}-cuda13.0-venvs}"],
+  "initializeCommand": ["/bin/bash", "-c", "mkdir -m 0755 -p ${localWorkspaceFolder}/../.{aws,cache,config/pip,local/share/${localWorkspaceFolderBasename}-cuda13.1-venvs}"],
   "postAttachCommand": ["/bin/bash", "-c", "if [ ${CODESPACES:-false} = 'true' ]; then . devcontainer-utils-post-attach-command; . rapids-post-attach-command; fi"],
   "workspaceFolder": "/home/coder",
   "workspaceMount": "source=${localWorkspaceFolder},target=/home/coder/cuvs,type=bind,consistency=consistent",
   "mounts": [
     "source=${localWorkspaceFolder}/../.aws,target=/home/coder/.aws,type=bind,consistency=consistent",
     "source=${localWorkspaceFolder}/../.cache,target=/home/coder/.cache,type=bind,consistency=consistent",
     "source=${localWorkspaceFolder}/../.config,target=/home/coder/.config,type=bind,consistency=consistent",
-    "source=${localWorkspaceFolder}/../.local/share/${localWorkspaceFolderBasename}-cuda13.0-venvs,target=/home/coder/.local/share/venvs,type=bind,consistency=consistent"
+    "source=${localWorkspaceFolder}/../.local/share/${localWorkspaceFolderBasename}-cuda13.1-venvs,target=/home/coder/.local/share/venvs,type=bind,consistency=consistent"
   ],
   "customizations": {
     "vscode": {
diff --git a/Dockerfile b/Dockerfile
@@ -1,3 +1,6 @@
+# SPDX-FileCopyrightText: Copyright (c) 2025-2026, NVIDIA CORPORATION. All rights reserved.
+# SPDX-License-Identifier: Apache-2.0
+
 # syntax=docker/dockerfile:1.5
 
 # ==============================================================================
@@ -55,7 +58,7 @@ RUN wget --quiet https://github.com/conda-forge/miniforge/releases/latest/downlo
 
 # Create conda environment and install cuVS packages with pinned dependencies
 RUN conda create -n cuvs python=${PYTHON_VER} -y && \
-    conda run -n cuvs conda install -c rapidsai -c conda-forge -c nvidia \
+    conda run -n cuvs conda install -c rapidsai -c conda-forge \
         cuvs=${RAPIDS_VER} \
         libcuvs=${RAPIDS_VER} \
         cuda-version=${CUDA_VER} \
diff --git a/README.md b/README.md
@@ -106,7 +106,7 @@ If installing a version that has not yet been released, the `rapidsai` channel c
 
 ```bash
 # CUDA 13
-conda install -c rapidsai-nightly -c conda-forge cuvs=26.02 cuda-version=13.0
+conda install -c rapidsai-nightly -c conda-forge cuvs=26.02 cuda-version=13.1
 
 # CUDA 12
 conda install -c rapidsai-nightly -c conda-forge cuvs=26.02 cuda-version=12.9
diff --git a/conda/environments/all_cuda-131_arch-aarch64.yaml b/conda/environments/all_cuda-131_arch-aarch64.yaml
@@ -16,7 +16,7 @@ dependencies:
 - cuda-nvtx-dev
 - cuda-profiler-api
 - cuda-python>=13.0.1,<14.0
-- cuda-version=13.0
+- cuda-version=13.1
 - cupy>=13.6.0
 - cxx-compiler
 - cython>=3.0.0,<3.2.0
@@ -53,4 +53,4 @@ dependencies:
 - sysroot_linux-aarch64==2.28
 - pip:
   - nvidia-sphinx-theme
-name: all_cuda-130_arch-aarch64
+name: all_cuda-131_arch-aarch64
diff --git a/conda/environments/all_cuda-131_arch-x86_64.yaml b/conda/environments/all_cuda-131_arch-x86_64.yaml
@@ -16,7 +16,7 @@ dependencies:
 - cuda-nvtx-dev
 - cuda-profiler-api
 - cuda-python>=13.0.1,<14.0
-- cuda-version=13.0
+- cuda-version=13.1
 - cupy>=13.6.0
 - cxx-compiler
 - cython>=3.0.0,<3.2.0
@@ -53,4 +53,4 @@ dependencies:
 - sysroot_linux-64==2.28
 - pip:
   - nvidia-sphinx-theme
-name: all_cuda-130_arch-x86_64
+name: all_cuda-131_arch-x86_64
diff --git a/conda/environments/bench_ann_cuda-131_arch-aarch64.yaml b/conda/environments/bench_ann_cuda-131_arch-aarch64.yaml
@@ -15,7 +15,7 @@ dependencies:
 - cuda-nvtx-dev
 - cuda-profiler-api
 - cuda-python>=13.0.1,<14.0
-- cuda-version=13.0
+- cuda-version=13.1
 - cupy>=13.6.0
 - cuvs==26.2.*,>=0.0.0a0
 - cxx-compiler
@@ -45,4 +45,4 @@ dependencies:
 - setuptools>=77.0.0
 - sysroot_linux-aarch64==2.28
 - wheel
-name: bench_ann_cuda-130_arch-aarch64
+name: bench_ann_cuda-131_arch-aarch64
diff --git a/conda/environments/bench_ann_cuda-131_arch-x86_64.yaml b/conda/environments/bench_ann_cuda-131_arch-x86_64.yaml
@@ -15,7 +15,7 @@ dependencies:
 - cuda-nvtx-dev
 - cuda-profiler-api
 - cuda-python>=13.0.1,<14.0
-- cuda-version=13.0
+- cuda-version=13.1
 - cupy>=13.6.0
 - cuvs==26.2.*,>=0.0.0a0
 - cxx-compiler
@@ -48,4 +48,4 @@ dependencies:
 - setuptools>=77.0.0
 - sysroot_linux-64==2.28
 - wheel
-name: bench_ann_cuda-130_arch-x86_64
+name: bench_ann_cuda-131_arch-x86_64
diff --git a/conda/environments/go_cuda-131_arch-aarch64.yaml b/conda/environments/go_cuda-131_arch-aarch64.yaml
@@ -14,7 +14,7 @@ dependencies:
 - cuda-nvcc
 - cuda-nvtx-dev
 - cuda-profiler-api
-- cuda-version=13.0
+- cuda-version=13.1
 - cxx-compiler
 - dlpack>=0.8,<1.0
 - gcc_linux-aarch64=14.*
@@ -29,4 +29,4 @@ dependencies:
 - nccl>=2.19
 - ninja
 - sysroot_linux-aarch64==2.28
-name: go_cuda-130_arch-aarch64
+name: go_cuda-131_arch-aarch64
diff --git a/conda/environments/go_cuda-131_arch-x86_64.yaml b/conda/environments/go_cuda-131_arch-x86_64.yaml
@@ -14,7 +14,7 @@ dependencies:
 - cuda-nvcc
 - cuda-nvtx-dev
 - cuda-profiler-api
-- cuda-version=13.0
+- cuda-version=13.1
 - cxx-compiler
 - dlpack>=0.8,<1.0
 - gcc_linux-64=14.*
@@ -29,4 +29,4 @@ dependencies:
 - nccl>=2.19
 - ninja
 - sysroot_linux-64==2.28
-name: go_cuda-130_arch-x86_64
+name: go_cuda-131_arch-x86_64
diff --git a/conda/environments/rust_cuda-131_arch-aarch64.yaml b/conda/environments/rust_cuda-131_arch-aarch64.yaml
@@ -13,7 +13,7 @@ dependencies:
 - cuda-nvcc
 - cuda-nvtx-dev
 - cuda-profiler-api
-- cuda-version=13.0
+- cuda-version=13.1
 - cxx-compiler
 - gcc_linux-aarch64=14.*
 - libclang==20.1.4
@@ -28,4 +28,4 @@ dependencies:
 - ninja
 - rust
 - sysroot_linux-aarch64==2.28
-name: rust_cuda-130_arch-aarch64
+name: rust_cuda-131_arch-aarch64
diff --git a/conda/environments/rust_cuda-131_arch-x86_64.yaml b/conda/environments/rust_cuda-131_arch-x86_64.yaml
@@ -13,7 +13,7 @@ dependencies:
 - cuda-nvcc
 - cuda-nvtx-dev
 - cuda-profiler-api
-- cuda-version=13.0
+- cuda-version=13.1
 - cxx-compiler
 - gcc_linux-64=14.*
 - libclang==20.1.4
@@ -28,4 +28,4 @@ dependencies:
 - ninja
 - rust
 - sysroot_linux-64==2.28
-name: rust_cuda-130_arch-x86_64
+name: rust_cuda-131_arch-x86_64
diff --git a/dependencies.yaml b/dependencies.yaml
@@ -3,7 +3,7 @@ files:
   all:
     output: conda
     matrix:
-      cuda: ["12.9", "13.0"]
+      cuda: ["12.9", "13.1"]
       arch: [x86_64, aarch64]
     includes:
       - build
@@ -30,7 +30,7 @@ files:
   bench_ann:
     output: conda
     matrix:
-      cuda: ["12.9", "13.0"]
+      cuda: ["12.9", "13.1"]
       arch: [x86_64, aarch64]
     includes:
       - bench
@@ -87,7 +87,7 @@ files:
   rust:
     output: conda
     matrix:
-      cuda: ["12.9", "13.0"]
+      cuda: ["12.9", "13.1"]
       arch: [x86_64, aarch64]
     includes:
       # clang/libclang only needed for bindgen support
@@ -102,7 +102,7 @@ files:
   go:
     output: conda
     matrix:
-      cuda: ["12.9", "13.0"]
+      cuda: ["12.9", "13.1"]
       arch: [x86_64, aarch64]
     includes:
       - clang
@@ -306,6 +306,10 @@ dependencies:
               cuda: "13.0"
             packages:
               - cuda-version=13.0
+          - matrix:
+              cuda: "13.1"
+            packages:
+              - cuda-version=13.1
   cuda:
     common:
       - output_types: [conda]
diff --git a/docs/source/build.rst b/docs/source/build.rst
@@ -46,7 +46,7 @@ C/C++ Package
 .. code-block:: bash
 
    # CUDA 13
-   conda install -c rapidsai -c conda-forge libcuvs cuda-version=13.0
+   conda install -c rapidsai -c conda-forge libcuvs cuda-version=13.1
 
    # CUDA 12
    conda install -c rapidsai -c conda-forge libcuvs cuda-version=12.9
@@ -57,7 +57,7 @@ Python Package
 .. code-block:: bash
 
    # CUDA 13
-   conda install -c rapidsai -c conda-forge cuvs cuda-version=13.0
+   conda install -c rapidsai -c conda-forge cuvs cuda-version=13.1
 
    # CUDA 12
    conda install -c rapidsai -c conda-forge cuvs cuda-version=12.9
@@ -98,7 +98,7 @@ Conda environment scripts are provided for installing the necessary dependencies
 
 .. code-block:: bash
 
-    conda env create --name cuvs -f conda/environments/all_cuda-130_arch-$(uname -m).yaml
+    conda env create --name cuvs -f conda/environments/all_cuda-131_arch-$(uname -m).yaml
     conda activate cuvs
 
 The recommended way to build and install cuVS from source is to use the `build.sh` script in the root of the repository. This script can build both the C++ and Python artifacts and provides CMake options for building and installing the headers, tests, benchmarks, and the pre-compiled shared library.
diff --git a/docs/source/cuvs_bench/index.rst b/docs/source/cuvs_bench/index.rst
@@ -65,7 +65,7 @@ Conda
    conda activate cuvs_benchmarks
 
    # to install GPU package:
-   conda install -c rapidsai -c conda-forge cuvs-bench=<rapids_version> cuda-version=13.0*
+   conda install -c rapidsai -c conda-forge cuvs-bench=<rapids_version> cuda-version=13.1*
 
    # to install CPU package for usage in CPU-only systems:
    conda install -c rapidsai -c conda-forge  cuvs-bench-cpu
diff --git a/examples/go/README.md b/examples/go/README.md
@@ -7,7 +7,7 @@ This package provides Go bindings for the cuVS (CUDA Vector Search) library.
 The required dependencies can be installed with a simple command (which creates your build environment):
 
 ```bash
-conda env create --name go -f conda/environments/go_cuda-130_arch-$(uname -m).yaml
+conda env create --name go -f conda/environments/go_cuda-131_arch-$(uname -m).yaml
 conda activate go
 ```
 You may prefer to use `mamba`, as it provides significant speedup over `conda`.
diff --git a/examples/rust/README.md b/examples/rust/README.md
@@ -9,7 +9,7 @@ Once the minimum requirements are satisfied, this example template application c
 You may follow these steps to quickly get set up:
 
 ```bash
-conda env create --name rust -f conda/environments/rust_cuda-130_arch-$(uname -m).yaml
+conda env create --name rust -f conda/environments/rust_cuda-131_arch-$(uname -m).yaml
 conda activate rust
 ```
 You may prefer to use `mamba`, as it provides significant speedup over `conda`.
