Skip to content

Comments

fix pytorch requirement for CUDA 13 builds#5244

Merged
rapids-bot[bot] merged 2 commits intorapidsai:branch-25.10from
jameslamb:fix/torch-req
Sep 4, 2025
Merged

fix pytorch requirement for CUDA 13 builds#5244
rapids-bot[bot] merged 2 commits intorapidsai:branch-25.10from
jameslamb:fix/torch-req

Conversation

@jameslamb
Copy link
Member

Follow-up to #5236

There, I intended to introduce a dependency on torch nightly wheels for CUDA 13, but made a typo in the dependencies.yaml filter. This fixes that, so that wheels for the correct torch variant + version will be installed in wheel-testing CI jobs here.

@jameslamb jameslamb requested a review from a team as a code owner September 2, 2025 22:54
@jameslamb jameslamb requested a review from msarahan September 2, 2025 22:54
@jameslamb jameslamb added improvement Improvement / enhancement to an existing function non-breaking Non-breaking change labels Sep 2, 2025
# TODO: remove this '.dev0' (which allows nightlies) once there are pytorch CUDA 13 released packages
# ref: https://github.com/pytorch/pytorch/issues/159779
- &pytorch_pip torch>=2.9.0a0
- &pytorch_pip torch>=2.9.0.dev0
Copy link
Member Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Guess this needs to be .dev instead of a0.

ERROR: Could not find a version that satisfies the requirement torch>=2.9.0a0 (from versions: 2.6.0, 2.7.0, 2.7.1, 2.8.0, 2.9.0.dev20250828+cu130, 2.9.0.dev20250829+cu130, 2.9.0.dev20250830+cu130, 2.9.0.dev20250831+cu130, 2.9.0.dev20250901+cu130, 2.9.0.dev20250902+cu130)

(pip devcontainer build link)

@jameslamb
Copy link
Member Author

/merge

@rapids-bot rapids-bot bot merged commit 9e79d5f into rapidsai:branch-25.10 Sep 4, 2025
101 checks passed
@jameslamb jameslamb deleted the fix/torch-req branch September 4, 2025 20:54
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

improvement Improvement / enhancement to an existing function non-breaking Non-breaking change

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants