Use custom test-infra branch

Author: NicolasHug

.github/workflows/linux_cuda_wheel.yaml

@@ -41,7 +41,7 @@ jobs:
     strategy:
       fail-fast: false
     name: Build and Upload wheel
-    uses: pytorch/test-infra/.github/workflows/build_wheels_linux.yml@main
+    uses: nicolashug/test-infra/.github/workflows/build_wheels_linux.yml@nvidia-driver
     with:
       repository: pytorch/torchcodec
       ref: ""