Update CuPy and NumPy dependency specs

[bdice]

Contributes to rapidsai/build-planning#279

RAPIDS is taking on requirements cupy>=14.0.1,!=14.1.0 and numpy>=2.0. Wheel dependencies on cupy-cuda12x and cupy-cuda13x now use the [ctk] extra.

See the linked issue for details.

[copy-pr-bot]

Auto-sync is disabled for draft pull requests in this repository. Workflows must be run manually.

Contributors can view more details about this message here.

[coderabbitai]

Note

Reviews paused

It looks like this branch is under active development. To avoid overwhelming you with review comments due to an influx of new commits, CodeRabbit has automatically paused this review. You can configure this behavior by changing the reviews.auto_review.auto_pause_after_reviewed_commits setting.

Use the following commands to manage reviews:

• @coderabbitai resume to resume automatic reviews.
• @coderabbitai review to trigger a single review.

Use the checkboxes below for quick actions:

• ▶️ Resume reviews
• 🔍 Trigger review

No actionable comments were generated in the recent review. 🎉

ℹ️ Recent review info

⚙️ Run configuration

Configuration used: Path: .coderabbit.yaml

Review profile: CHILL

Plan: Enterprise

Run ID: 9b882ba1-cdb8-4389-a601-7f615af85aaa

📥 Commits

Reviewing files that changed from the base of the PR and between 6d3eba0 and f6cb5a5.

📒 Files selected for processing (1)

• dependencies.yaml

🚧 Files skipped from review as they are similar to previous changes (1)

• dependencies.yaml

---

📝 Walkthrough

Summary by CodeRabbit

• Chores
  • Updated dependency version constraints to require NumPy >=2.0,<3.0 across build/test setups, shared dependencies, and CUDA conda environments (including x86_64 and aarch64 variants).
  • Raised CuPy minimum versions to >=14.0.1 across supported configurations and adjusted exclusions accordingly, while keeping 14.1.0 excluded.

Walkthrough

The PR raises NumPy requirements to >=2.0 across project, conda, and environment definitions, and updates CuPy minimum version constraints to >=14.0.1 in the recipe and dependency matrix.

Changes

Dependency version updates

Layer / File(s)	Summary
NumPy minimum version update dependencies.yaml, python/rmm/pyproject.toml, conda/environments/*	NumPy constraints are raised to >=2.0 across shared runtime, test matrix, project metadata, and CUDA environment definitions.
CuPy minimum version update conda/recipes/rmm/recipe.yaml, dependencies.yaml	CuPy constraints for cupy, cupy-cuda12x, and cupy-cuda13x are raised to >=14.0.1, and the 14.0.0 exclusion is removed.

Estimated code review effort

🎯 2 (Simple) | ⏱️ ~10 minutes

Possibly related PRs

• rapidsai/rmm#2240: Updates CuPy-related dependency groups in the same dependency matrix area.
• rapidsai/rmm#2427: Modifies the same CuPy version constraints in dependencies.yaml and conda/recipes/rmm/recipe.yaml.

Suggested reviewers

• jameslamb
• jakirkham

🚥 Pre-merge checks | ✅ 5

✅ Passed checks (5 passed)

Check name	Status	Explanation
Title check	✅ Passed	The title clearly summarizes the main change: updating CuPy and NumPy dependency specifications.
Description check	✅ Passed	The description directly matches the dependency updates and linked build-planning issue.
Docstring Coverage	✅ Passed	No functions found in the changed files to evaluate docstring coverage. Skipping docstring coverage check.
Linked Issues check	✅ Passed	Check skipped because no linked issues were found for this pull request.
Out of Scope Changes check	✅ Passed	Check skipped because no linked issues were found for this pull request.

✨ Finishing Touches

🧪 Generate unit tests (beta)

• Create PR with unit tests

---

_{Comment @coderabbitai help to get the list of available commands.}

[jameslamb]

I pulled in latest main. PR CI was not running:

[Invalid workflow file: .github/workflows/pr.yaml#L397](https://github.com/rapidsai/rmm/actions/runs/28244160524/workflow)
error parsing called workflow
".github/workflows/pr.yaml"
-> "rapidsai/shared-workflows/.github/workflows/build-in-devcontainer.yaml@cuda-13.3.0"
: failed to fetch workflow: reference to workflow should be either a valid branch, tag, or commit

https://github.com/rapidsai/rmm/actions/runs/28244160524

Needed #2453

[jameslamb]

Left one question about a follow-up, otherwise this all looks good to me. Thanks for the fixes for the [ctk] extra.

[jameslamb]

This is ready to merge, but holding off because cudf isn't ready for these pins yet: NVIDIA/cudf#22962 (review)

Putting them here would force them onto cudf.

[gforsyth]

/merge