File tree Expand file tree Collapse file tree 2 files changed +6
-6
lines changed Expand file tree Collapse file tree 2 files changed +6
-6
lines changed Original file line number Diff line number Diff line change @@ -6,7 +6,7 @@ PLATFORM=$(python -c "import sys; print(sys.platform)")
6
6
7
7
# Install all the dependencies required for Torch-TensorRT
8
8
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
10
10
11
11
# Install Torch-TensorRT
12
12
if [[ ${PLATFORM} == win32 ]]; then
Original file line number Diff line number Diff line change 1
- REM Source: https://github.com/pytorch/builder/blob/4e109742d88ff3c85e77d60bc4d90d229d7f6afe/windows/internal/driver_update.bat
2
1
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
5
5
if errorlevel 1 exit /b 1
6
6
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
8
8
if errorlevel 1 exit /b 1
9
9
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
You can’t perform that action at this time.
0 commit comments