43 lines
1.1 KiB
YAML
43 lines
1.1 KiB
YAML
services:
|
|
ultralytics:
|
|
image: ultralytics/ultralytics:latest
|
|
container_name: ultralytics
|
|
restart: unless-stopped
|
|
command:
|
|
- /bin/bash
|
|
- -c
|
|
- >-
|
|
pip install 'streamlit>=1.29.0' -q --root-user-action=ignore &&
|
|
sed -i 's/cv2.destroyAllWindows()/pass/' /ultralytics/ultralytics/solutions/streamlit_inference.py &&
|
|
streamlit run /ultralytics/ultralytics/solutions/streamlit_inference.py
|
|
--server.headless true --server.address 0.0.0.0 --server.port 8501
|
|
-- /data/models/yolo11x_leaf.pt
|
|
ports:
|
|
# - "127.0.0.1:8501:8501"
|
|
- "8501:8501"
|
|
volumes:
|
|
- /srv/ultralytics/data:/data
|
|
- /srv/ultralytics/runs:/root/runs
|
|
- pip_cache:/root/.cache/pip
|
|
deploy:
|
|
resources:
|
|
reservations:
|
|
devices:
|
|
- driver: nvidia
|
|
count: all
|
|
capabilities: [gpu]
|
|
environment:
|
|
- TZ=America/New_York
|
|
- STREAMLIT_SERVER_ADDRESS=0.0.0.0
|
|
- STREAMLIT_SERVER_PORT=8501
|
|
- STREAMLIT_SERVER_HEADLESS=true
|
|
networks:
|
|
- npm-network
|
|
|
|
networks:
|
|
npm-network:
|
|
external: true
|
|
|
|
volumes:
|
|
pip_cache:
|