Hugging Face's logo

Spaces:
yeq6x
/
Image2Body_gradio
Running on Zero

App Files Community
yeq6x commited on
Commit
6090dd8
·
1 Parent(s): 145c501

.

Browse files
Files changed (1) hide show
  1. docker-compose.yml +4 -0
docker-compose.yml CHANGED
@@ -14,3 +14,7 @@ services:
14
  - capabilities: [ "gpu" ]
15
  count: 1
16
  driver: nvidia
 
 
 
 
 
14
  - capabilities: [ "gpu" ]
15
  count: 1
16
  driver: nvidia
17
+ runtime: nvidia
18
+ environment:
19
+ - NVIDIA_VISIBLE_DEVICES=all
20
+ - NVIDIA_DRIVER_CAPABILITIES=compute,utility