Skip to content

Commit bfec9c7

Browse files
authored
CUDA 13: use releases, not nightlies, for PyTorch wheels (#5356)
Contributes to rapidsai/build-planning#208 Follow-up to #5236 Similar to rapidsai/cugraph-gnn#355 PyTorch now has official releases with CUDA 13 support. This switches from nightlies to those releases for wheels in CI here. Authors: - James Lamb (https://github.com/jameslamb) Approvers: - Bradley Dice (https://github.com/bdice) URL: #5356
1 parent b94f8a9 commit bfec9c7

1 file changed

Lines changed: 2 additions & 6 deletions

File tree

dependencies.yaml

Lines changed: 2 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -461,9 +461,7 @@ dependencies:
461461
- matrix:
462462
cuda: "13.*"
463463
packages:
464-
# TODO: switch from nightlies to releases when there are pytorch CUDA 13 released packages
465-
# ref: https://github.com/pytorch/pytorch/issues/159779
466-
- --extra-index-url=https://download.pytorch.org/whl/nightly/cu130
464+
- --extra-index-url=https://download.pytorch.org/whl/cu130
467465
- {matrix: null, packages: null}
468466
- output_types: [requirements, pyproject]
469467
matrices:
@@ -475,9 +473,7 @@ dependencies:
475473
- matrix:
476474
cuda: "13.*"
477475
packages:
478-
# TODO: remove this '.dev0' (which allows nightlies) once there are pytorch CUDA 13 released packages
479-
# ref: https://github.com/pytorch/pytorch/issues/159779
480-
- &pytorch_pip torch>=2.9.0.dev0
476+
- &pytorch_pip torch>=2.9.0
481477
- &tensordict tensordict>=0.1.2
482478
- {matrix: null, packages: [*pytorch_pip, *tensordict]}
483479
depends_on_libcudf:

0 commit comments

Comments
 (0)