Spaces:

yeq6x
/

Image2Body_gradio

Runtime error

yeq6x commited on Dec 23, 2024

Commit

6090dd8

1 Parent(s): 145c501

.

Files changed (1) hide show

docker-compose.yml CHANGED Viewed

@@ -14,3 +14,7 @@ services:
             - capabilities: [ "gpu" ]
               count: 1
               driver: nvidia

             - capabilities: [ "gpu" ]
               count: 1
               driver: nvidia
+    runtime: nvidia
+    environment:
+      - NVIDIA_VISIBLE_DEVICES=all
+      - NVIDIA_DRIVER_CAPABILITIES=compute,utility