File size: 800 Bytes
286f32d
5447961
 
592da45
5447961
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
286f32d
5447961
286f32d
 
 
 
 
 
5447961
286f32d
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
services:
  app:
    build:
       dockerfile: Dockerfile.nvidia
       target: app
    ports:
       - 8000:7860
    env_file:
       - .env
    volumes:
       - ./yologp:/app/yologp
       - ./assets:/app/assets
       - ./data:/app/data
    deploy:
       resources:
         reservations:
           devices:
             - driver: nvidia
               count: 1
               capabilities: [ gpu ]
  # hf_app:
  #   build:
  #     dockerfile: Dockerfile.hf
  #     target: app
  #   ports:
  #     - 8000:7860
  #   env_file:
  #     - .env
  #   volumes:
  #     - ./inference_gradio_app.py:/app/inference_gradio_app.py
  #   deploy:
  #     resources:
  #       reservations:
  #         devices:
  #           - driver: nvidia
  #             count: 1
  #             capabilities: [ gpu ]