From 933ea6a1fc313e31629f6809d8be744ce783f7a5 Mon Sep 17 00:00:00 2001
From: moto <855818+mthrok@users.noreply.github.com>
Date: Wed, 4 Sep 2024 14:53:47 +0900
Subject: [PATCH] test NVIDIA_DRIVER_CAPABILITIES

---
 .github/workflows/conda_cuda_test.yml | 3 +++
 .github/workflows/wheel_cuda_test.yml | 3 +++
 2 files changed, 6 insertions(+)

diff --git a/.github/workflows/conda_cuda_test.yml b/.github/workflows/conda_cuda_test.yml
index 4a425754..c9c80ba0 100644
--- 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
index 3537ed50..2c993fc9 100644
--- 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 }})"