Fix: CI

Aivis-Project · Nov 8, 2024 · a50b464 · a50b464
1 parent 333e17b
commit a50b464
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/.github/workflows/build-engine-container.yml b/.github/workflows/build-engine-container.yml
@@ -58,7 +58,7 @@ jobs:
           - tag: nvidia
             target: runtime-nvidia-env
             base_image: ubuntu:20.04
-            base_runtime_image: nvidia/cuda:11.8.0-cudnn8-runtime-ubuntu20.04
+            base_runtime_image: nvidia/cuda:12.6.2-cudnn8-runtime-ubuntu20.04
             platforms: linux/amd64
 
     steps: