Compare commits
2
Commits
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
89d93a5a20 | ||
|
|
9ae1cd9918 |
@@ -0,0 +1,20 @@
|
||||
# VideoDetect — Docker Compose environment variables
|
||||
# Copy this file to .env and fill in real values before running docker-compose up.
|
||||
|
||||
# -----------------------------------------------------
|
||||
# Database
|
||||
# -----------------------------------------------------
|
||||
DB_ROOT_PASSWORD=changeme_root
|
||||
DB_PASSWORD=changeme_videodetect
|
||||
|
||||
# -----------------------------------------------------
|
||||
# Storage paths (host-side mounts)
|
||||
# -----------------------------------------------------
|
||||
NAS_OUTPUT_PATH=${PWD}/output
|
||||
MODELS_PATH=${PWD}/models
|
||||
TRAINING_PATH=${PWD}/training
|
||||
|
||||
# -----------------------------------------------------
|
||||
# Grafana
|
||||
# -----------------------------------------------------
|
||||
GRAFANA_PASSWORD=changeme_grafana
|
||||
+11
-8
@@ -1,5 +1,3 @@
|
||||
version: '3.8'
|
||||
|
||||
services:
|
||||
mariadb:
|
||||
image: mariadb:10.11
|
||||
@@ -29,11 +27,10 @@ services:
|
||||
|
||||
worker:
|
||||
build:
|
||||
context: ./worker
|
||||
dockerfile: Dockerfile
|
||||
context: .
|
||||
dockerfile: worker/Dockerfile
|
||||
container_name: videodetect-worker
|
||||
restart: unless-stopped
|
||||
runtime: nvidia
|
||||
environment:
|
||||
- NVIDIA_VISIBLE_DEVICES=all
|
||||
- NVIDIA_DRIVER_CAPABILITIES=compute,utility
|
||||
@@ -45,7 +42,7 @@ services:
|
||||
- CONFIG_PATH=/app/config.yaml
|
||||
volumes:
|
||||
- /dev/null:/dev/null # tmpfs mounted at /scratch in container
|
||||
- ${NAS_INPUT_PATH:-/mnt/nas/input}:/data/input:ro
|
||||
- nas_input:/data/input:ro
|
||||
- ${NAS_OUTPUT_PATH:-/mnt/nas/output}:/data/output
|
||||
- ${MODELS_PATH:-/mnt/nas/models}:/models
|
||||
- ${TRAINING_PATH:-/mnt/nas/training}:/data/training
|
||||
@@ -57,7 +54,7 @@ services:
|
||||
devices:
|
||||
- driver: nvidia
|
||||
count: all
|
||||
capabilities: [gpu]
|
||||
capabilities: ["gpu"]
|
||||
limits:
|
||||
memory: 24G
|
||||
networks:
|
||||
@@ -86,7 +83,7 @@ services:
|
||||
- FLASK_ENV=production
|
||||
volumes:
|
||||
- ./ui:/app
|
||||
- ${NAS_INPUT_PATH:-/mnt/nas/input}:/data/input:ro
|
||||
- nas_input:/data/input:ro
|
||||
- ${NAS_OUTPUT_PATH:-/mnt/nas/output}:/data/output
|
||||
networks:
|
||||
- videodetect-network
|
||||
@@ -142,6 +139,12 @@ volumes:
|
||||
driver: local
|
||||
grafana_data:
|
||||
driver: local
|
||||
nas_input:
|
||||
driver: local
|
||||
driver_opts:
|
||||
type: nfs
|
||||
o: addr=10.0.0.2,ro,nfsvers=4,hard,intr
|
||||
device: ":/mnt/Bulk/Homes/ryan/Prawns"
|
||||
|
||||
networks:
|
||||
videodetect-network:
|
||||
|
||||
+2
-1
@@ -1,5 +1,6 @@
|
||||
# Base image: CUDA 11.8 runtime on Ubuntu 22.04
|
||||
FROM nvidia/cuda:11.8.0-runtime-ubuntu22.04
|
||||
#FROM nvidia/cuda:11.8.0-runtime-ubuntu22.04
|
||||
FROM nvidia/cuda:13.3.1-cudnn-runtime-ubuntu22.04
|
||||
|
||||
# Avoid interactive prompts during build
|
||||
ENV DEBIAN_FRONTEND=noninteractive
|
||||
|
||||
Reference in New Issue
Block a user