test NVIDIA_DRIVER_CAPABILITIES

facebookresearch · Sep 4, 2024 · 933ea6a · 933ea6a
1 parent 38c879e
commit 933ea6a
Showing 2 changed files with 6 additions and 0 deletions.
diff --git a/.github/workflows/conda_cuda_test.yml b/.github/workflows/conda_cuda_test.yml
@@ -16,6 +16,9 @@ on:
         required: true
         type: string
 
+env:
+  NVIDIA_DRIVER_CAPABILITIES: all
+
 jobs:
   unit-test:
     name: "Unit test (ffmpeg ${{ matrix.ffmpeg-version }})"

diff --git a/.github/workflows/wheel_cuda_test.yml b/.github/workflows/wheel_cuda_test.yml
@@ -16,6 +16,9 @@ on:
         required: true
         type: string
 
+env:
+  NVIDIA_DRIVER_CAPABILITIES: all
+
 jobs:
   unit-test:
     name: "Unit test (ffmpeg ${{ matrix.ffmpeg-version }})"