update

Author: bhimrazy

.lightning/workflows/pytorch.yml

@@ -123,8 +123,7 @@ run: |
   extra=$(python -c "print({'lightning': 'pytorch-'}.get('${PACKAGE_NAME}', ''))")
 
   # Use find-links to prefer CUDA-specific packages from PyTorch index
-  uv pip install -e ".[${extra}dev]" --upgrade \
-    --find-links https://download.pytorch.org/whl/cu${CUDA_VERSION_MM}
+  uv pip install -e ".[${extra}dev]" --upgrade
 
   echo "Ensure only a single package is installed"
   if [ "${PACKAGE_NAME}" == "pytorch" ]; then

requirements/pytorch/test.txt

@@ -21,5 +21,6 @@ uvicorn  # for `ServableModuleValidator`  # not setting version as re-defined in
 
 tensorboard >=2.11, <2.21.0  # for `TensorBoardLogger`
 
-torch-tensorrt; platform_system == "Linux" and python_version >= "3.12"
+# TODO: resolve GPU test failures for TensorRT due to defaulting to cu13 installations
+torch-tensorrt<2.9.0; platform_system == "Linux" and python_version >= "3.12"
 huggingface-hub