Update docker-compose.yaml

add capabilities: [gpu] to enable gpu on window docker desktop
taiyouZhang · Dec 14, 2023 · 355b2cb · 355b2cb
1 parent b3ea59c
commit 355b2cb
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/docker-compose.yaml b/docker-compose.yaml
@@ -30,4 +30,5 @@ services:
         reservations:
           devices:
             - driver: nvidia
+              capabilities: [gpu]
               device_ids: ['all']