From b50fd2283ff1e0e3eb8a3987b960487a2ef2aa71 Mon Sep 17 00:00:00 2001 From: Bradley Dice Date: Wed, 24 Jun 2026 00:40:46 -0500 Subject: [PATCH 1/4] Update CuPy and NumPy dependency specs. --- conda/environments/all_cuda-129_arch-aarch64.yaml | 4 ++-- conda/environments/all_cuda-129_arch-x86_64.yaml | 4 ++-- conda/environments/all_cuda-133_arch-aarch64.yaml | 4 ++-- conda/environments/all_cuda-133_arch-x86_64.yaml | 4 ++-- conda/recipes/pylibraft/recipe.yaml | 4 ++-- conda/recipes/raft-dask/recipe.yaml | 2 +- dependencies.yaml | 8 ++++---- python/pylibraft/pyproject.toml | 4 ++-- 8 files changed, 17 insertions(+), 17 deletions(-) diff --git a/conda/environments/all_cuda-129_arch-aarch64.yaml b/conda/environments/all_cuda-129_arch-aarch64.yaml index e031bb0e1e..eb65106dbb 100644 --- a/conda/environments/all_cuda-129_arch-aarch64.yaml +++ b/conda/environments/all_cuda-129_arch-aarch64.yaml @@ -15,7 +15,7 @@ dependencies: - cuda-profiler-api - cuda-python>=12.9.2,<13.0 - cuda-version=12.9 -- cupy>=13.6.0,!=14.0.0,!=14.1.0 +- cupy>=14.0.1,!=14.1.0 - cxx-compiler - cython>=3.2.2 - dask-cuda==26.8.*,>=0.0.0a0 @@ -31,7 +31,7 @@ dependencies: - libucxx==0.51.*,>=0.0.0a0 - nccl>=2.19 - ninja -- numpy>=1.23,<3.0 +- numpy>=2.0,<3.0 - numpydoc - pre-commit - pylibraft==26.8.*,>=0.0.0a0 diff --git a/conda/environments/all_cuda-129_arch-x86_64.yaml b/conda/environments/all_cuda-129_arch-x86_64.yaml index 524d462636..7efe10afd0 100644 --- a/conda/environments/all_cuda-129_arch-x86_64.yaml +++ b/conda/environments/all_cuda-129_arch-x86_64.yaml @@ -15,7 +15,7 @@ dependencies: - cuda-profiler-api - cuda-python>=12.9.2,<13.0 - cuda-version=12.9 -- cupy>=13.6.0,!=14.0.0,!=14.1.0 +- cupy>=14.0.1,!=14.1.0 - cxx-compiler - cython>=3.2.2 - dask-cuda==26.8.*,>=0.0.0a0 @@ -31,7 +31,7 @@ dependencies: - libucxx==0.51.*,>=0.0.0a0 - nccl>=2.19 - ninja -- numpy>=1.23,<3.0 +- numpy>=2.0,<3.0 - numpydoc - pre-commit - pylibraft==26.8.*,>=0.0.0a0 diff --git a/conda/environments/all_cuda-133_arch-aarch64.yaml b/conda/environments/all_cuda-133_arch-aarch64.yaml index eea90d2ff4..46fb9f72ac 100644 --- a/conda/environments/all_cuda-133_arch-aarch64.yaml +++ b/conda/environments/all_cuda-133_arch-aarch64.yaml @@ -15,7 +15,7 @@ dependencies: - cuda-profiler-api - cuda-python>=13.0.1,<14.0 - cuda-version=13.3 -- cupy>=13.6.0,!=14.0.0,!=14.1.0 +- cupy>=14.0.1,!=14.1.0 - cxx-compiler - cython>=3.2.2 - dask-cuda==26.8.*,>=0.0.0a0 @@ -31,7 +31,7 @@ dependencies: - libucxx==0.51.*,>=0.0.0a0 - nccl>=2.19 - ninja -- numpy>=1.23,<3.0 +- numpy>=2.0,<3.0 - numpydoc - pre-commit - pylibraft==26.8.*,>=0.0.0a0 diff --git a/conda/environments/all_cuda-133_arch-x86_64.yaml b/conda/environments/all_cuda-133_arch-x86_64.yaml index 25e8abe3c6..8121267477 100644 --- a/conda/environments/all_cuda-133_arch-x86_64.yaml +++ b/conda/environments/all_cuda-133_arch-x86_64.yaml @@ -15,7 +15,7 @@ dependencies: - cuda-profiler-api - cuda-python>=13.0.1,<14.0 - cuda-version=13.3 -- cupy>=13.6.0,!=14.0.0,!=14.1.0 +- cupy>=14.0.1,!=14.1.0 - cxx-compiler - cython>=3.2.2 - dask-cuda==26.8.*,>=0.0.0a0 @@ -31,7 +31,7 @@ dependencies: - libucxx==0.51.*,>=0.0.0a0 - nccl>=2.19 - ninja -- numpy>=1.23,<3.0 +- numpy>=2.0,<3.0 - numpydoc - pre-commit - pylibraft==26.8.*,>=0.0.0a0 diff --git a/conda/recipes/pylibraft/recipe.yaml b/conda/recipes/pylibraft/recipe.yaml index 28f095f5c1..3373fd706b 100644 --- a/conda/recipes/pylibraft/recipe.yaml +++ b/conda/recipes/pylibraft/recipe.yaml @@ -89,13 +89,13 @@ requirements: - libraft =${{ version }} - libraft-headers =${{ version }} - python - - numpy >=1.23,<3.0 + - numpy >=2.0,<3.0 - rmm =${{ minor_version }} - if: cuda_major == "12" then: cuda-python >=12.9.2,<13.0 else: cuda-python >=13.0.1,<14.0 run_constraints: - - cupy >=13.6.0,!=14.0.0,!=14.1.0 + - cupy >=14.0.1,!=14.1.0 ignore_run_exports: from_package: - ${{ compiler("c") }} diff --git a/conda/recipes/raft-dask/recipe.yaml b/conda/recipes/raft-dask/recipe.yaml index 282dec6331..dae5b73dec 100644 --- a/conda/recipes/raft-dask/recipe.yaml +++ b/conda/recipes/raft-dask/recipe.yaml @@ -99,7 +99,7 @@ requirements: then: cuda-python >=12.9.2,<13.0 else: cuda-python >=13.0.1,<14.0 run_constraints: - - cupy >=13.6.0,!=14.0.0,!=14.1.0 + - cupy >=14.0.1,!=14.1.0 ignore_run_exports: from_package: - ${{ compiler("c") }} diff --git a/dependencies.yaml b/dependencies.yaml index 43ac510c2f..3bcb53febe 100644 --- a/dependencies.yaml +++ b/dependencies.yaml @@ -367,7 +367,7 @@ dependencies: common: - output_types: conda packages: - - cupy>=13.6.0,!=14.0.0,!=14.1.0 + - cupy>=14.0.1,!=14.1.0 # NOTE: This is intentionally not broken into groups by a 'cuda_suffixed' selector like # other packages with -cu{nn}x suffixes in this file. # All RAPIDS wheel builds (including in devcontainers) expect cupy to be suffixed. @@ -377,11 +377,11 @@ dependencies: - matrix: cuda: "12.*" packages: - - cupy-cuda12x>=13.6.0,!=14.0.0,!=14.1.0 + - cupy-cuda12x[ctk]>=14.0.1,!=14.1.0 # fallback to CUDA 13 versions if 'cuda' is '13.*' or not provided - matrix: packages: - - cupy-cuda13x>=13.6.0,!=14.0.0,!=14.1.0 + - cupy-cuda13x[ctk]>=14.0.1,!=14.1.0 depends_on_rapids_logger: common: - output_types: [conda, requirements, pyproject] @@ -438,7 +438,7 @@ dependencies: common: - output_types: [conda, pyproject, requirements] packages: - - numpy>=1.23,<3.0 + - numpy>=2.0,<3.0 run_raft_dask: common: - output_types: [conda, pyproject, requirements] diff --git a/python/pylibraft/pyproject.toml b/python/pylibraft/pyproject.toml index 345162788d..7df18bc2c1 100644 --- a/python/pylibraft/pyproject.toml +++ b/python/pylibraft/pyproject.toml @@ -22,7 +22,7 @@ requires-python = ">=3.11" dependencies = [ "cuda-python>=13.0.1,<14.0", "libraft==26.8.*,>=0.0.0a0", - "numpy>=1.23,<3.0", + "numpy>=2.0,<3.0", "rmm==26.8.*,>=0.0.0a0", ] # This list was generated by `rapids-dependency-file-generator`. To make changes, edit ../../dependencies.yaml and run `rapids-dependency-file-generator`. classifiers = [ @@ -36,7 +36,7 @@ classifiers = [ [project.optional-dependencies] test = [ - "cupy-cuda13x>=13.6.0,!=14.0.0,!=14.1.0", + "cupy-cuda13x[ctk]>=14.0.1,!=14.1.0", "pytest", "pytest-cov", "scikit-learn", From 03379404a2757ff17cce47a3a0a9903b58ba5acc Mon Sep 17 00:00:00 2001 From: Bradley Dice Date: Thu, 25 Jun 2026 00:01:25 -0500 Subject: [PATCH 2/4] Fix CuPy constraints for wheel tests --- dependencies.yaml | 12 +++++++++++- 1 file changed, 11 insertions(+), 1 deletion(-) diff --git a/dependencies.yaml b/dependencies.yaml index 3bcb53febe..dc0fc42cfe 100644 --- a/dependencies.yaml +++ b/dependencies.yaml @@ -372,7 +372,17 @@ dependencies: # other packages with -cu{nn}x suffixes in this file. # All RAPIDS wheel builds (including in devcontainers) expect cupy to be suffixed. specific: - - output_types: [requirements, pyproject] + - output_types: requirements + matrices: + - matrix: + cuda: "12.*" + packages: + - cupy-cuda12x>=14.0.1,!=14.1.0 + # fallback to CUDA 13 versions if 'cuda' is '13.*' or not provided + - matrix: + packages: + - cupy-cuda13x>=14.0.1,!=14.1.0 + - output_types: pyproject matrices: - matrix: cuda: "12.*" From beb60f2be9accdae0831ca4e1db4fba0ff727dd2 Mon Sep 17 00:00:00 2001 From: Bradley Dice Date: Thu, 25 Jun 2026 09:47:54 -0500 Subject: [PATCH 3/4] Guard CuPy CTK extra for wheel metadata --- dependencies.yaml | 10 ++++++++++ 1 file changed, 10 insertions(+) diff --git a/dependencies.yaml b/dependencies.yaml index dc0fc42cfe..71073704ff 100644 --- a/dependencies.yaml +++ b/dependencies.yaml @@ -384,6 +384,16 @@ dependencies: - cupy-cuda13x>=14.0.1,!=14.1.0 - output_types: pyproject matrices: + - matrix: + cuda: "12.*" + use_cuda_wheels: "false" + packages: + - cupy-cuda12x>=14.0.1,!=14.1.0 + - matrix: + cuda: "13.*" + use_cuda_wheels: "false" + packages: + - cupy-cuda13x>=14.0.1,!=14.1.0 - matrix: cuda: "12.*" packages: From d6a63e1367941c9e67f966caf4aaae57c02eecb3 Mon Sep 17 00:00:00 2001 From: Bradley Dice Date: Fri, 26 Jun 2026 17:35:20 -0500 Subject: [PATCH 4/4] Reuse CuPy dependency anchors --- dependencies.yaml | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/dependencies.yaml b/dependencies.yaml index 71073704ff..0d7a739bb4 100644 --- a/dependencies.yaml +++ b/dependencies.yaml @@ -377,23 +377,23 @@ dependencies: - matrix: cuda: "12.*" packages: - - cupy-cuda12x>=14.0.1,!=14.1.0 + - &cupy_cu12 cupy-cuda12x>=14.0.1,!=14.1.0 # fallback to CUDA 13 versions if 'cuda' is '13.*' or not provided - matrix: packages: - - cupy-cuda13x>=14.0.1,!=14.1.0 + - &cupy_cu13 cupy-cuda13x>=14.0.1,!=14.1.0 - output_types: pyproject matrices: - matrix: cuda: "12.*" use_cuda_wheels: "false" packages: - - cupy-cuda12x>=14.0.1,!=14.1.0 + - *cupy_cu12 - matrix: cuda: "13.*" use_cuda_wheels: "false" packages: - - cupy-cuda13x>=14.0.1,!=14.1.0 + - *cupy_cu13 - matrix: cuda: "12.*" packages: