调整显存占用

2024-07-30 12:03:56 +08:00
parent 74ebbc414a
commit 2a24fa358d
2 changed files with 11 additions and 4 deletions
--- a/docker-compose.yml
+++ b/docker-compose.yml
@@ -108,7 +108,7 @@ services:
      resources:
        reservations:
          devices:
-            - device_ids: [ "1" ]
+            - device_ids: [ "0", "1" ]
              capabilities: [ "gpu" ]
              driver: "nvidia"

@@ -126,7 +126,7 @@ services:
      resources:
        reservations:
          devices:
-            - device_ids: [ "1" ]
+            - device_ids: [ "0", "1" ]
              capabilities: [ "gpu" ]
              driver: "nvidia"

@@ -144,6 +144,6 @@ services:
      resources:
        reservations:
          devices:
-            - device_ids: [ "1" ]
+            - device_ids: [ "0", "1" ]
              capabilities: [ "gpu" ]
              driver: "nvidia"