Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
2 changes: 1 addition & 1 deletion .github/workflows/build.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -62,7 +62,7 @@ jobs:
branch: ${{ inputs.branch }}
arch: "${{matrix.arch}}"
date: ${{ inputs.date }}
container_image: "rapidsai/ci-wheel:26.04-cuda${{ matrix.cuda_version }}-rockylinux8-py3.10"
container_image: "rapidsai/ci-wheel:26.04-cuda${{ matrix.cuda_version }}-rockylinux8-py3.11"
node_type: "cpu16"
requires_license_builder: true
script: "ci/build_standalone_c.sh"
Expand Down
4 changes: 2 additions & 2 deletions .github/workflows/pr.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -194,7 +194,7 @@ jobs:
build_type: pull-request
arch: "${{matrix.arch}}"
date: ${{ inputs.date }}_c
container_image: "rapidsai/ci-wheel:26.04-cuda${{ matrix.cuda_version }}-rockylinux8-py3.10"
container_image: "rapidsai/ci-wheel:26.04-cuda${{ matrix.cuda_version }}-rockylinux8-py3.11"
node_type: "cpu16"
requires_license_builder: true
script: "ci/build_standalone_c.sh --build-tests"
Expand All @@ -221,7 +221,7 @@ jobs:
node_type: "gpu-l4-latest-1"
arch: "${{matrix.arch}}"
date: ${{ inputs.date }}_c
container_image: "rapidsai/ci-wheel:26.04-cuda${{ matrix.cuda_version }}-rockylinux8-py3.10"
container_image: "rapidsai/ci-wheel:26.04-cuda${{ matrix.cuda_version }}-rockylinux8-py3.11"
script: "ci/test_standalone_c.sh libcuvs_c_${{ matrix.cuda_version }}_${{ matrix.arch }}.tar.gz"
sha: ${{ inputs.sha }}
conda-java-build-and-tests:
Expand Down
2 changes: 1 addition & 1 deletion Dockerfile
Original file line number Diff line number Diff line change
Expand Up @@ -17,7 +17,7 @@
# Configurable build arguments - override with --build-arg during build
ARG CUDA_VER=12.9.1

# Python version for the conda environment (supported: 3.10, 3.11, 3.12)
# Python version for the conda environment (supported: 3.11, 3.12, 3.13)
ARG PYTHON_VER=3.12

# RAPIDS/cuVS version
Expand Down
6 changes: 1 addition & 5 deletions dependencies.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -410,10 +410,6 @@ dependencies:
specific:
- output_types: conda
matrices:
- matrix:
py: "3.10"
packages:
- python=3.10
- matrix:
py: "3.11"
packages:
Expand All @@ -428,7 +424,7 @@ dependencies:
- python=3.13
- matrix:
packages:
- python>=3.10,<3.14
- python>=3.11,<3.14
run_py_cuvs:
common:
- output_types: [conda, requirements, pyproject]
Expand Down
12 changes: 6 additions & 6 deletions docs/source/cuvs_bench/index.rst
Original file line number Diff line number Diff line change
Expand Up @@ -84,21 +84,21 @@ We provide images for GPU enabled systems, as well as systems without a GPU. The

Nightly images are located in `dockerhub <https://hub.docker.com/r/rapidsai/cuvs-bench/tags>`_.

The following command pulls the nightly container for Python version 3.10, CUDA version 12.5, and cuVS version 24.12:
The following command pulls the nightly container for Python version 3.13, CUDA version 12.9, and cuVS version 26.04:

.. code-block:: bash

docker pull rapidsai/cuvs-bench:26.04a-cuda12.5-py3.10 # substitute cuvs-bench for the exact desired container.
docker pull rapidsai/cuvs-bench:26.04a-cuda12-py3.13 # substitute cuvs-bench for the exact desired container.

The CUDA and python versions can be changed for the supported values:
- Supported CUDA versions: 12
- Supported Python versions: 3.10 and 3.11.
- Supported CUDA versions: 12, 13
- Supported Python versions: 3.11, 3.12, and 3.13

You can see the exact versions as well in the dockerhub site:
- `cuVS bench images <https://hub.docker.com/r/rapidsai/cuvs-bench/tags>`_
- `cuVS bench CPU only images <https://hub.docker.com/r/rapidsai/cuvs-bench-cpu/tags>`_

**Note:** GPU containers use the CUDA toolkit from inside the container, the only requirement is a driver installed on the host machine that supports that version. So, for example, CUDA 11.8 containers can run in systems with a CUDA 12.x capable driver. Please also note that the Nvidia-Docker runtime from the `Nvidia Container Toolkit <https://github.com/NVIDIA/nvidia-docker>`_ is required to use GPUs inside docker containers.
**Note:** GPU containers use the CUDA toolkit from inside the container, the only requirement is a driver installed on the host machine that supports that version. So, for example, CUDA 12 containers can run in systems with a CUDA 13.x capable driver. Please also note that the Nvidia-Docker runtime from the `Nvidia Container Toolkit <https://github.com/NVIDIA/nvidia-docker>`_ is required to use GPUs inside docker containers.

Running the benchmarks
======================
Expand Down Expand Up @@ -277,7 +277,7 @@ The container arguments in the above section also be used for the CPU-only conta
export DATA_FOLDER=path/to/store/datasets/and/results
docker run --rm -it -u $(id -u) \
-v $DATA_FOLDER:/data/benchmarks \
rapidsai/cuvs-bench-cpu:24.10a-py3.10 \
rapidsai/cuvs-bench-cpu:26.04a-py3.13 \
"--dataset deep-image-96-angular" \
"--normalize" \
"--algorithms hnswlib --batch-size 10 -k 10" \
Expand Down
5 changes: 2 additions & 3 deletions python/cuvs/pyproject.toml
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
# SPDX-FileCopyrightText: Copyright (c) 2024-2025, NVIDIA CORPORATION.
# SPDX-FileCopyrightText: Copyright (c) 2024-2026, NVIDIA CORPORATION.
# SPDX-License-Identifier: Apache-2.0

[build-system]
Expand All @@ -18,7 +18,7 @@ authors = [
{ name = "NVIDIA Corporation" },
]
license = "Apache-2.0"
requires-python = ">=3.10"
requires-python = ">=3.11"
dependencies = [
"cuda-python>=13.0.1,<14.0",
"libcuvs==26.4.*,>=0.0.0a0",
Expand All @@ -28,7 +28,6 @@ dependencies = [
classifiers = [
"Intended Audience :: Developers",
"Programming Language :: Python",
"Programming Language :: Python :: 3.10",
"Programming Language :: Python :: 3.11",
"Programming Language :: Python :: 3.12",
"Programming Language :: Python :: 3.13",
Expand Down
5 changes: 2 additions & 3 deletions python/cuvs_bench/pyproject.toml
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
# SPDX-FileCopyrightText: Copyright (c) 2024-2025, NVIDIA CORPORATION.
# SPDX-FileCopyrightText: Copyright (c) 2024-2026, NVIDIA CORPORATION.
# SPDX-License-Identifier: Apache-2.0

[build-system]
Expand All @@ -17,7 +17,7 @@ authors = [
{ name = "NVIDIA Corporation" },
]
license = "Apache-2.0"
requires-python = ">=3.10"
requires-python = ">=3.11"
dependencies = [
"click",
"cuvs==26.4.*,>=0.0.0a0",
Expand All @@ -32,7 +32,6 @@ classifiers = [
"Topic :: Database",
"Topic :: Scientific/Engineering",
"Programming Language :: Python",
"Programming Language :: Python :: 3.10",
"Programming Language :: Python :: 3.11",
"Programming Language :: Python :: 3.12",
"Programming Language :: Python :: 3.13",
Expand Down
2 changes: 1 addition & 1 deletion python/libcuvs/pyproject.toml
Original file line number Diff line number Diff line change
Expand Up @@ -17,7 +17,7 @@ authors = [
{ name = "NVIDIA Corporation" },
]
license = "Apache-2.0"
requires-python = ">=3.10"
requires-python = ">=3.11"
dependencies = [
"cuda-toolkit[cublas,curand,cusolver,cusparse]>=12,<14",
"libraft==26.4.*,>=0.0.0a0",
Expand Down