.

Author: ahmadsharif1

.github/workflows/linux_cuda_wheel.yaml

@@ -63,6 +63,8 @@ jobs:
         python-version: ['3.9']
         cuda-version: ['12.4']
         ffmpeg-version-for-tests: ['5.1.2', '6.1.1', '7.0.1']
+    container:
+      image: "pytorch/manylinux-builder:cuda${{ matrix.cuda-version }}"
     if: ${{ always() }}
     needs: build
     steps: