调整显卡调用

2024-08-02 09:54:06 +08:00
parent 45a2289b99
commit 8a56308b03
2 changed files with 4 additions and 4 deletions
--- a/docker-compose.yml
+++ b/docker-compose.yml
@@ -108,7 +108,7 @@ services:
      resources:
        reservations:
          devices:
-            - device_ids: [ "0", "1" ]
+            - device_ids: [ "1" ]
              capabilities: [ "gpu" ]
              driver: "nvidia"

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

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