fix: Missing nvinfer.dll in CI for Windows [release/2.3] (#2884)

Author: gs-olive

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

@@ -9,4 +9,4 @@ ${CONDA_RUN} pip install --pre -r ${PWD}/tests/py/requirements.txt --use-depreca
 # Install Torch-TensorRT via pre-built wheels. On windows, the location of wheels is not fixed.
 ${CONDA_RUN} pip install ${RUNNER_ARTIFACT_DIR}/torch_tensorrt*.whl
 
-echo -e "Running test script";
+echo -e "Running test script";

packaging/pre_build_script_windows.sh

@@ -1,7 +1,8 @@
 python -m pip install -U numpy packaging pyyaml setuptools wheel
 
 # Install TRT 10 from PyPi
-python -m pip install tensorrt==10.0.1 --extra-index-url https://pypi.nvidia.com
+TRT_VERSION=$(python -c "import yaml; print(yaml.safe_load(open('dev_dep_versions.yml', 'r'))['__tensorrt_version__'])")
+pip install tensorrt==${TRT_VERSION} tensorrt-cu12-bindings==${TRT_VERSION} tensorrt-cu12-libs==${TRT_VERSION} --extra-index-url https://pypi.nvidia.com
 
 choco install bazelisk -y
 

tests/py/requirements.txt

@@ -6,9 +6,11 @@ pytest-xdist>=3.6.1
 networkx==2.8.8
 torch==2.3.0
 torchvision==0.18.0
---extra-index-url https://pypi.ngc.nvidia.com
+--extra-index-url https://pypi.nvidia.com
 pyyaml
 tensorrt==10.0.1
+tensorrt-cu12-bindings==10.0.1
+tensorrt-cu12-libs==10.0.1
 timm>=1.0.3
 transformers==4.39.3
 parameterized>=0.2.0