Skip to content

Commit f83f6ae

Browse files
Switch to release channel for PyTorch + CUDA 13 (#355)
Switches from the nightly channel to the release channel for PyTorch with CUDA 13. --------- Co-authored-by: Gil Forsyth <[email protected]>
1 parent 4005ab1 commit f83f6ae

File tree

3 files changed

+5
-9
lines changed

3 files changed

+5
-9
lines changed

ci/test_wheel_cugraph-pyg.sh

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -20,7 +20,7 @@ CUDA_MAJOR="${RAPIDS_CUDA_VERSION%%.*}"
2020
if [[ "${CUDA_MAJOR}" == "12" ]]; then
2121
PYTORCH_INDEX="https://download.pytorch.org/whl/cu126"
2222
else
23-
PYTORCH_INDEX="https://download.pytorch.org/whl/nightly/cu130"
23+
PYTORCH_INDEX="https://download.pytorch.org/whl/cu130"
2424
fi
2525

2626
# echo to expand wildcard before adding `[extra]` requires for pip

dependencies.yaml

Lines changed: 2 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -377,9 +377,7 @@ dependencies:
377377
- matrix:
378378
cuda: "13.*"
379379
packages:
380-
# TODO: switch from nightlies to releases when there are pytorch CUDA 13 released packages
381-
# ref: https://github.com/pytorch/pytorch/issues/159779
382-
- --extra-index-url=https://download.pytorch.org/whl/nightly/cu130
380+
- --extra-index-url=https://download.pytorch.org/whl/cu130
383381
- matrix:
384382
packages:
385383
- output_types: [requirements, pyproject]
@@ -391,9 +389,7 @@ dependencies:
391389
- matrix:
392390
cuda: "13.*"
393391
packages:
394-
# TODO: remove this 'dev0' (which allows nightlies) once there are pytorch CUDA 13 released packages
395-
# ref: https://github.com/pytorch/pytorch/issues/159779
396-
- &pytorch_pip torch>=2.9.0.dev0
392+
- &pytorch_pip torch>=2.9.0
397393
- matrix:
398394
packages:
399395
- *pytorch_pip

python/cugraph-pyg/pyproject.toml

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
1-
# SPDX-FileCopyrightText: Copyright (c) 2022-2024, NVIDIA CORPORATION.
1+
# SPDX-FileCopyrightText: Copyright (c) 2022-2025, NVIDIA CORPORATION.
22
# SPDX-License-Identifier: Apache-2.0
33

44
[build-system]
@@ -55,7 +55,7 @@ test = [
5555
"pytest-cov",
5656
"pytest-xdist",
5757
"sentence-transformers",
58-
"torch>=2.9.0.dev0",
58+
"torch>=2.9.0",
5959
] # This list was generated by `rapids-dependency-file-generator`. To make changes, edit ../../dependencies.yaml and run `rapids-dependency-file-generator`.
6060

6161
[tool.setuptools]

0 commit comments

Comments
 (0)