chore: move external dep installation into a separate script

.github/scripts/install-external-deps.sh

@@ -0,0 +1,24 @@
+#!/bin/bash
+
+# This script installs all the external dependencies required for Torch-TensorRT
+# It defines the required variables and installs the dependencies
+
+set -x
+
+# Set default values for CHANNEL and CU_VERSION if not defined
+CHANNEL=${CHANNEL:-nightly}
+CU_VERSION=${CU_VERSION:-cu129}
+
+# Define variables for torch and torchvision versions
+TORCH=$(grep "^torch>" ${PWD}/py/requirements.txt)
+TORCHVISION=$(grep "^torchvision>" ${PWD}/tests/py/requirements.txt)
+INDEX_URL=https://download.pytorch.org/whl/${CHANNEL}/${CU_VERSION}
+
+# Install all the dependencies required for Torch-TensorRT
+pip install --pre -r ${PWD}/tests/py/requirements.txt
+# dependencies in the tests/py/requirements.txt might install a different version of torch or torchvision
+# eg. timm will install the latest torchvision, however we want to use the torchvision from nightly
+# reinstall torch torchvision to make sure we have the correct version
+pip uninstall -y torch torchvision
+pip install --force-reinstall --pre ${TORCHVISION} --index-url ${INDEX_URL}
+pip install --force-reinstall --pre ${TORCH} --index-url ${INDEX_URL}

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

@@ -1,9 +1,6 @@
 #set -exou pipefail
 set -x
 
-TORCH=$(grep "^torch>" ${PWD}/py/requirements.txt)
-TORCHVISION=$(grep "^torchvision>" ${PWD}/tests/py/requirements.txt)
-INDEX_URL=https://download.pytorch.org/whl/${CHANNEL}/${CU_VERSION}
 PLATFORM=$(python -c "import sys; print(sys.platform)")
 
 if [[ $(uname -m) == "aarch64" ]]; then
@@ -12,14 +9,8 @@ if [[ $(uname -m) == "aarch64" ]]; then
     install_cuda_aarch64
 fi
 
-# Install all the dependencies required for Torch-TensorRT
-pip install --pre -r ${PWD}/tests/py/requirements.txt
-# dependencies in the tests/py/requirements.txt might install a different version of torch or torchvision
-# eg. timm will install the latest torchvision, however we want to use the torchvision from nightly
-# reinstall torch torchvisionto make sure we have the correct version
-pip uninstall -y torch torchvision
-pip install --force-reinstall --pre ${TORCHVISION} --index-url ${INDEX_URL}
-pip install --force-reinstall --pre ${TORCH} --index-url ${INDEX_URL}
+# Install all the external dependencies required for Torch-TensorRT
+source .github/scripts/install-external-deps.sh
 
 
 # Install Torch-TensorRT

tests/py/requirements.txt

@@ -14,5 +14,5 @@ nvidia-modelopt[all]; python_version >'3.9' and python_version <'3.13'
 # flashinfer-python is broken on python 3.9 at the moment, so skip it for now
 flashinfer-python; python_version >'3.9' and python_version <'3.13'
 --extra-index-url https://download.pytorch.org/whl/nightly/cu129
-torchvision>=0.23.0.dev,<0.24.0
+torchvision>=0.24.0.dev,<0.25.0
 timm>=1.0.3