Add CI testing job with CV-CUDA support

.github/scripts/setup-env.sh

@@ -82,6 +82,12 @@ echo '::group::Install TorchVision'
 pip install -e . -v --no-build-isolation
 echo '::endgroup::'
 
+if [[ "${CVCUDA:-}" == "1" ]]; then
+  echo '::group::Install CV-CUDA'
+  pip install --progress-bar=off cvcuda-cu12
+  echo '::endgroup::'
+fi
+
 echo '::group::Install torchvision-extra-decoders'
 # This can be done after torchvision was built
 if [[ "$(uname)" == "Linux" && "$(uname -m)" != "aarch64" ]]; then

.github/workflows/tests.yml

@@ -45,6 +45,36 @@ jobs:
 
         ./.github/scripts/unittest.sh
 
+  unittests-linux-cvcuda:
+    strategy:
+      matrix:
+        python-version:
+          - "3.10"
+        runner: ["linux.g5.4xlarge.nvidia.gpu"]
+        gpu-arch-type: ["cuda"]
+        gpu-arch-version: ["12.6"]
+      fail-fast: false
+    uses: pytorch/test-infra/.github/workflows/linux_job_v2.yml@main
+    permissions:
+      id-token: write
+      contents: read
+    with:
+      repository: pytorch/vision
+      runner: ${{ matrix.runner }}
+      gpu-arch-type: ${{ matrix.gpu-arch-type }}
+      gpu-arch-version: ${{ matrix.gpu-arch-version }}
+      timeout: 120
+      test-infra-ref: main
+      script: |
+        set -euo pipefail
+
+        export PYTHON_VERSION=${{ matrix.python-version }}
+        export GPU_ARCH_TYPE=${{ matrix.gpu-arch-type }}
+        export GPU_ARCH_VERSION=${{ matrix.gpu-arch-version }}
+        export CVCUDA="1"
+
+        ./.github/scripts/unittest.sh
+
   unittests-macos:
     strategy:
       matrix: