28 lines
566 B
YAML
28 lines
566 B
YAML
|
version: "3.8"
|
||
|
|
|
||
|
|
services:
|
||
|
|
app:
|
||
|
|
build:
|
||
|
|
context: .
|
||
|
|
dockerfile: Dockerfile
|
||
|
|
volumes:
|
||
|
|
- .:/app
|
||
|
|
- ./data:/app/data
|
||
|
|
ports:
|
||
|
|
- "8000:8000"
|
||
|
|
environment:
|
||
|
|
- PYTHONUNBUFFERED=1
|
||
|
|
command: poetry run python main.py
|
||
|
|
|
||
|
|
# TODO: uncomment code below if you want to run it on GPU
|
||
|
|
# environment:
|
||
|
|
# - NVIDIA_VISIBLE_DEVICES=all
|
||
|
|
# deploy:
|
||
|
|
# mode: replicated
|
||
|
|
# replicas: 1
|
||
|
|
# resources:
|
||
|
|
# reservations:
|
||
|
|
# devices:
|
||
|
|
# - driver: nvidia
|
||
|
|
# device_ids: [ '0' ]
|
||
|
|
# capabilities: [ gpu ]
|