Skip to content

Commit bad4c98

Browse files
authored
[None][infra] use public torch index as CI backup (NVIDIA#12261)
Signed-off-by: Tyler Burt <195370667+tburt-nv@users.noreply.github.com>
1 parent 4c4b688 commit bad4c98

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

jenkins/L0_Test.groovy

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -3655,9 +3655,9 @@ def launchTestJobs(pipeline, testFilter)
36553655
// Use internal mirror instead of https://download.pytorch.org/whl/cu130 for better network stability.
36563656
// PyTorch CUDA 13.0 package and torchvision package can be installed as expected.
36573657
if (k8s_arch == "amd64") {
3658-
trtllm_utils.llmExecStepWithRetry(pipeline, script: "pip3 install torch==2.9.1+cu130 torchvision==0.24.1+cu130 --extra-index-url https://urm.nvidia.com/artifactory/api/pypi/pytorch-cu128-remote/simple")
3658+
trtllm_utils.llmExecStepWithRetry(pipeline, script: "pip3 install torch==2.9.1+cu130 torchvision==0.24.1+cu130 --extra-index-url https://urm.nvidia.com/artifactory/api/pypi/pytorch-cu128-remote/simple --extra-index-url https://download.pytorch.org/whl/cu130")
36593659
} else {
3660-
trtllm_utils.llmExecStepWithRetry(pipeline, script: "pip3 install torch==2.9.1+cu130 torchvision==0.24.1 --extra-index-url https://urm.nvidia.com/artifactory/api/pypi/pytorch-cu128-remote/simple")
3660+
trtllm_utils.llmExecStepWithRetry(pipeline, script: "pip3 install torch==2.9.1+cu130 torchvision==0.24.1 --extra-index-url https://urm.nvidia.com/artifactory/api/pypi/pytorch-cu128-remote/simple --extra-index-url https://download.pytorch.org/whl/cu130")
36613661
}
36623662
}
36633663

0 commit comments

Comments
 (0)