Migrate dockerfile to gpu spec for docker v27 (#748)

zoriya · Jan 2, 2025 · f8b587a · f8b587a
1 parent e4f7c12
commit f8b587a
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 0 deletions.
diff --git a/docker-compose.build.yml b/docker-compose.build.yml
@@ -122,6 +122,9 @@ services:
         reservations:
           devices:
             - capabilities: [gpu]
+              driver: cdi
+              device_ids:
+                - nvidia.com/gpu=all
     environment:
       - GOCODER_PREFIX=/video
       - GOCODER_HWACCEL=nvidia

diff --git a/docker-compose.dev.yml b/docker-compose.dev.yml
@@ -150,6 +150,9 @@ services:
         reservations:
           devices:
             - capabilities: [gpu]
+              driver: cdi
+              device_ids:
+                - nvidia.com/gpu=all
     environment:
       - GOCODER_PREFIX=/video
       - GOCODER_HWACCEL=nvidia

diff --git a/docker-compose.yml b/docker-compose.yml
@@ -104,6 +104,9 @@ services:
         reservations:
           devices:
             - capabilities: [gpu]
+              driver: cdi
+              device_ids:
+                - nvidia.com/gpu=all
     environment:
       - GOCODER_PREFIX=/video
       - GOCODER_HWACCEL=nvidia