.

ahmadsharif1 · ahmadsharif1 · commit 7730c4b5de40 · 2024-10-21T13:59:57.000-07:00
diff --git a/.github/workflows/linux_cuda_wheel.yaml b/.github/workflows/linux_cuda_wheel.yaml
@@ -66,6 +66,7 @@ jobs:
         ffmpeg-version-for-tests: ['6']
     container:
       image: "pytorch/manylinux-builder:cuda${{ matrix.cuda-version }}"
+      options: "--gpus all"
     if: ${{ always() }}
     needs: build
     steps:
