| version: "3.8" | |
| services: | |
| rvc: | |
| build: | |
| context: . | |
| dockerfile: Dockerfile | |
| container_name: rvc | |
| volumes: | |
| - ./weights:/app/assets/weights | |
| - ./opt:/app/opt | |
| # - ./dataset:/app/dataset # you can use this folder in order to provide your dataset for model training | |
| ports: | |
| - 7865:7865 | |
| deploy: | |
| resources: | |
| reservations: | |
| devices: | |
| - driver: nvidia | |
| count: 1 | |
| capabilities: [gpu] |