.

Author: ahmadsharif1

.github/workflows/linux_cuda_wheel.yaml

@@ -66,7 +66,7 @@ jobs:
         ffmpeg-version-for-tests: ['6']
     container:
       image: "pytorch/manylinux-builder:cuda${{ matrix.cuda-version }}"
-      options: "--gpus all"
+      options: "--gpus all,capabilities=all"
     if: ${{ always() }}
     needs: build
     steps: