Use deploy resources instead of runtime nvidia

emsi · Jul 4, 2023 · 42502c9 · 42502c9
1 parent 7a5a192
commit 42502c9
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/docker/docker-compose.yml b/docker/docker-compose.yml
@@ -1,6 +1,12 @@
 services:
  stable-diffusion-webui:
- runtime: nvidia
+ deploy:
+ resources:
+ reservations:
+ devices:
+ - driver: nvidia
+ count: 1
+ capabilities: [gpu]
  image: emsi/stable-diffusion-webui
  build:
  context: ../