Skip to content

Commit a95f9dd

Browse files
[None][infra] Cherry pick fix for 1.2.0rc6.post4 (#12867)
Signed-off-by: Tyler Burt <195370667+tburt-nv@users.noreply.github.com> Signed-off-by: Yiteng Niu <6831097+niukuo@users.noreply.github.com> Signed-off-by: Yuanjing Xue <197832395+yuanjingx87@users.noreply.github.com> Co-authored-by: tburt-nv <195370667+tburt-nv@users.noreply.github.com>
1 parent 87dd01a commit a95f9dd

1 file changed

Lines changed: 2 additions & 2 deletions

File tree

jenkins/L0_Test.groovy

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -3201,9 +3201,9 @@ def launchTestJobs(pipeline, testFilter)
32013201
// Use internal mirror instead of https://download.pytorch.org/whl/cu130 for better network stability.
32023202
// PyTorch CUDA 13.0 package and torchvision package can be installed as expected.
32033203
if (k8s_arch == "amd64") {
3204-
trtllm_utils.llmExecStepWithRetry(pipeline, script: "pip3 install torch==2.9.0+cu130 torchvision==0.24.0+cu130 --extra-index-url https://urm.nvidia.com/artifactory/api/pypi/pytorch-cu128-remote/simple")
3204+
trtllm_utils.llmExecStepWithRetry(pipeline, script: "pip3 install torch==2.9.0+cu130 torchvision==0.24.0+cu130 --extra-index-url https://urm.nvidia.com/artifactory/api/pypi/pytorch-cu128-remote/simple --extra-index-url https://download.pytorch.org/whl/cu130")
32053205
} else {
3206-
trtllm_utils.llmExecStepWithRetry(pipeline, script: "pip3 install torch==2.9.0+cu130 torchvision==0.24.0 --extra-index-url https://urm.nvidia.com/artifactory/api/pypi/pytorch-cu128-remote/simple")
3206+
trtllm_utils.llmExecStepWithRetry(pipeline, script: "pip3 install torch==2.9.0+cu130 torchvision==0.24.0 --extra-index-url https://urm.nvidia.com/artifactory/api/pypi/pytorch-cu128-remote/simple --extra-index-url https://download.pytorch.org/whl/cu130")
32073207
}
32083208
}
32093209

0 commit comments

Comments
 (0)