reverse requirements and ci.

Author: GdoongMathew

.github/workflows/ci-tests-pytorch.yml

@@ -140,8 +140,7 @@ jobs:
             -U --upgrade-strategy=eager --prefer-binary \
             -r requirements/_integrations/accelerators.txt \
             -r requirements/_integrations/tensorrt.txt \
-            --extra-index-url="${TORCH_URL}" --find-links="${PYPI_CACHE_DIR}" \
-            --extra-index-url https://download.pytorch.org/whl/torch-tensorrt
+            --extra-index-url="${TORCH_URL}" --find-links="${PYPI_CACHE_DIR}"
           pip list
       - name: Drop LAI from extensions
         if: ${{ matrix.pkg-name != 'lightning' }}

requirements/_integrations/tensorrt.txt

@@ -18,3 +18,6 @@ fastapi  # for `ServableModuleValidator`  # not setting version as re-defined in
 uvicorn  # for `ServableModuleValidator`  # not setting version as re-defined in App
 
 tensorboard >=2.9.1, <2.21.0  # for `TensorBoardLogger`
+
+--extra-index-url https://download.pytorch.org/whl/torch-tensorrt
+torch-tensorrt >= 2.3.0; platform_system == "Linux" and python_version >= "3.12"  # Limit the torch-tensorrt version for testing.