diff --git a/jenkins/L0_Test.groovy b/jenkins/L0_Test.groovy index 4bbb700deea..1b6100ac41b 100644 --- a/jenkins/L0_Test.groovy +++ b/jenkins/L0_Test.groovy @@ -2995,7 +2995,13 @@ def launchTestJobs(pipeline, testFilter) // Extra PyTorch CUDA 13.0 install for all bare-metal environments (Default PyTorch is for CUDA 12.8) if (values[6]) { echo "###### Extra PyTorch CUDA 13.0 install Start ######" - trtllm_utils.llmExecStepWithRetry(pipeline, script: "pip3 install torch==2.9.0 torchvision --index-url https://download.pytorch.org/whl/cu130") + // Use internal mirror instead of https://download.pytorch.org/whl/cu130 for better network stability. + // PyTorch CUDA 13.0 package and torchvision package can be installed as expected. + if (k8s_arch == "amd64") { + 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") + } else { + 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") + } } def libEnv = []