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