From 2c2685cb60e0ebd4e3b1d3a2968f53ad1798f2f4 Mon Sep 17 00:00:00 2001 From: lanluo-nvidia Date: Mon, 2 Jun 2025 16:45:42 -0700 Subject: [PATCH] cherry pick 3505 to release2.6.1 --- .github/scripts/install-torch-tensorrt.sh | 2 +- packaging/driver_upgrade.bat | 10 +++++----- 2 files changed, 6 insertions(+), 6 deletions(-) diff --git a/.github/scripts/install-torch-tensorrt.sh b/.github/scripts/install-torch-tensorrt.sh index 86dfeda343..c75bfdb09d 100644 --- a/.github/scripts/install-torch-tensorrt.sh +++ b/.github/scripts/install-torch-tensorrt.sh @@ -6,7 +6,7 @@ PLATFORM=$(python -c "import sys; print(sys.platform)") # Install all the dependencies required for Torch-TensorRT pip install --pre ${TORCH_TORCHVISION} --index-url ${INDEX_URL} -pip install --pre -r ${PWD}/tests/py/requirements.txt --use-deprecated legacy-resolver +pip install --pre -r ${PWD}/tests/py/requirements.txt # Install Torch-TensorRT if [[ ${PLATFORM} == win32 ]]; then diff --git a/packaging/driver_upgrade.bat b/packaging/driver_upgrade.bat index 886e908f55..352005d69f 100644 --- a/packaging/driver_upgrade.bat +++ b/packaging/driver_upgrade.bat @@ -1,10 +1,10 @@ -REM Source: https://github.com/pytorch/builder/blob/4e109742d88ff3c85e77d60bc4d90d229d7f6afe/windows/internal/driver_update.bat -set "DRIVER_DOWNLOAD_LINK=https://ossci-windows.s3.amazonaws.com/528.89-data-center-tesla-desktop-winserver-2016-2019-2022-dch-international.exe" -curl --retry 3 -kL %DRIVER_DOWNLOAD_LINK% --output 528.89-data-center-tesla-desktop-winserver-2016-2019-2022-dch-international.exe +set WIN_DRIVER_VN=528.89 +set "DRIVER_DOWNLOAD_LINK=https://ossci-windows.s3.amazonaws.com/%WIN_DRIVER_VN%-data-center-tesla-desktop-winserver-2016-2019-2022-dch-international.exe" +curl --retry 3 -kL %DRIVER_DOWNLOAD_LINK% --output %WIN_DRIVER_VN%-data-center-tesla-desktop-winserver-2016-2019-2022-dch-international.exe if errorlevel 1 exit /b 1 -start /wait 528.89-data-center-tesla-desktop-winserver-2016-2019-2022-dch-international.exe -s -noreboot +start /wait %WIN_DRIVER_VN%-data-center-tesla-desktop-winserver-2016-2019-2022-dch-international.exe -s -noreboot if errorlevel 1 exit /b 1 -del 528.89-data-center-tesla-desktop-winserver-2016-2019-2022-dch-international.exe || ver > NUL +del %WIN_DRIVER_VN%-data-center-tesla-desktop-winserver-2016-2019-2022-dch-international.exe || ver > NUL