Skip to content

Commit 707f16d

Browse files
cherry pick 3505(windows driver upgrade) to release2.6.1 (#3547)
1 parent 505b002 commit 707f16d

File tree

2 files changed

+6
-6
lines changed

2 files changed

+6
-6
lines changed

.github/scripts/install-torch-tensorrt.sh

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,7 @@ PLATFORM=$(python -c "import sys; print(sys.platform)")
66

77
# Install all the dependencies required for Torch-TensorRT
88
pip install --pre ${TORCH_TORCHVISION} --index-url ${INDEX_URL}
9-
pip install --pre -r ${PWD}/tests/py/requirements.txt --use-deprecated legacy-resolver
9+
pip install --pre -r ${PWD}/tests/py/requirements.txt
1010

1111
# Install Torch-TensorRT
1212
if [[ ${PLATFORM} == win32 ]]; then

packaging/driver_upgrade.bat

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -1,10 +1,10 @@
1-
REM Source: https://github.com/pytorch/builder/blob/4e109742d88ff3c85e77d60bc4d90d229d7f6afe/windows/internal/driver_update.bat
21

3-
set "DRIVER_DOWNLOAD_LINK=https://ossci-windows.s3.amazonaws.com/528.89-data-center-tesla-desktop-winserver-2016-2019-2022-dch-international.exe"
4-
curl --retry 3 -kL %DRIVER_DOWNLOAD_LINK% --output 528.89-data-center-tesla-desktop-winserver-2016-2019-2022-dch-international.exe
2+
set WIN_DRIVER_VN=528.89
3+
set "DRIVER_DOWNLOAD_LINK=https://ossci-windows.s3.amazonaws.com/%WIN_DRIVER_VN%-data-center-tesla-desktop-winserver-2016-2019-2022-dch-international.exe"
4+
curl --retry 3 -kL %DRIVER_DOWNLOAD_LINK% --output %WIN_DRIVER_VN%-data-center-tesla-desktop-winserver-2016-2019-2022-dch-international.exe
55
if errorlevel 1 exit /b 1
66

7-
start /wait 528.89-data-center-tesla-desktop-winserver-2016-2019-2022-dch-international.exe -s -noreboot
7+
start /wait %WIN_DRIVER_VN%-data-center-tesla-desktop-winserver-2016-2019-2022-dch-international.exe -s -noreboot
88
if errorlevel 1 exit /b 1
99

10-
del 528.89-data-center-tesla-desktop-winserver-2016-2019-2022-dch-international.exe || ver > NUL
10+
del %WIN_DRIVER_VN%-data-center-tesla-desktop-winserver-2016-2019-2022-dch-international.exe || ver > NUL

0 commit comments

Comments
 (0)