File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff 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
You can’t perform that action at this time.
0 commit comments