File tree Expand file tree Collapse file tree 3 files changed +5
-9
lines changed
Expand file tree Collapse file tree 3 files changed +5
-9
lines changed Original file line number Diff line number Diff line change @@ -20,7 +20,7 @@ CUDA_MAJOR="${RAPIDS_CUDA_VERSION%%.*}"
2020if [[ " ${CUDA_MAJOR} " == " 12" ]]; then
2121 PYTORCH_INDEX=" https://download.pytorch.org/whl/cu126"
2222else
23- PYTORCH_INDEX=" https://download.pytorch.org/whl/nightly/ cu130"
23+ PYTORCH_INDEX=" https://download.pytorch.org/whl/cu130"
2424fi
2525
2626# echo to expand wildcard before adding `[extra]` requires for pip
Original file line number Diff line number Diff 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
Original file line number Diff line number Diff line change 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 ]
You can’t perform that action at this time.
0 commit comments