Add gpu-arch-type (try 4)

Author: pearu

.github/workflows/docs.yaml

@@ -15,17 +15,18 @@ defaults:
 
 jobs:
   build-docs:
-    runs-on: linux.g5.4xlarge.nvidia.gpu
     strategy:
       fail-fast: false
       matrix:
         python-version: ['3.10']
         cuda-version: ['12.6']
         ffmpeg-version: ['7']
+    uses: pytorch/test-infra/.github/workflows/linux_job_v2.yml@main
     with:
+      runner: linux.g5.4xlarge.nvidia.gpu
       gpu-arch-type: cuda
       gpu-arch-version: "12.6"
-    steps:
+      steps:
       - name: Setup env vars
         run: |
           cuda_version_without_periods=$(echo "${{ matrix.cuda-version }}" | sed 's/\.//g')