Update test_cli.yml

Bebra777228 · Aug 2, 2024 · 097a90f · 097a90f
1 parent f4a29e7
commit 097a90f
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/.github/workflows/test_cli.yml b/.github/workflows/test_cli.yml
@@ -12,6 +12,10 @@ jobs:
   test:
     runs-on: ubuntu-latest
 
+    container:
+      image: nvidia/cuda:11.0-base
+      options: --gpus 1
+
     steps:
     - name: Checkout code
       uses: actions/checkout@v2
@@ -28,8 +32,8 @@ jobs:
 
     - name: Install ffmpeg
       run: |
-        sudo apt-get update
-        sudo apt-get install -y ffmpeg
+        apt-get update
+        apt-get install -y ffmpeg
 
     - name: Install assets
       run: |