fix: update GPU resource reservations in docker-compose files to use CDI driver

2026-05-22 16:06:51 +00:00 · 2025-03-10 04:05:56 +08:00
parent 0fc8c7ab07
commit 1fb6393649
2 changed files with 28 additions and 9 deletions
@@ -1,11 +1,23 @@
 services:
-  test:
-    image: nvidia/cuda:12.3.1-base-ubuntu20.04
-    command: nvidia-smi
+  testgpupodman_count:
+    image: docker.io/ubuntu:latest
+    command: ["nvidia-smi"]
+    profiles: [gpu]
    deploy:
      resources:
        reservations:
          devices:
-            - driver: nvidia
-              count: 1
-              capabilities: [gpu]
+          - driver: nvidia
+            count: 1
+            capabilities: [gpu]
+  test:
+    image: docker.io/ubuntu:latest
+    command: ["nvidia-smi"]
+    deploy:
+      resources:
+        reservations:
+          devices:
+          - driver: cdi
+            count: 1
+            device_ids: ['nvidia.com/gpu=all']
+            capabilities: [gpu]