.

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,\"capabilities=video\""
+      options: "--gpus all -e NVIDIA_DRIVER_CAPABILITIES=video,compute,utility"
     if: ${{ always() }}
     needs: build
     steps: