Add Docker environment setup

This commit is contained in:
JayXiao
2026-03-17 14:20:26 +08:00
parent 0a3e6e19e6
commit 3b9ca418fa
3 changed files with 72 additions and 0 deletions

21
docker-compose.yml Normal file
View File

@@ -0,0 +1,21 @@
services:
jupyter:
build: .
container_name: gie-eris-jupyter
ports:
- "${JUPYTER_PORT:-8888}:8888"
volumes:
- ./notebooks:/workspace/notebooks
- ./media-pipeline:/workspace/media-pipeline
- ./scripts:/workspace/scripts
- ${HF_CACHE_DIR:-./.cache/huggingface}:/workspace/.cache/huggingface
env_file:
- .env
deploy:
resources:
reservations:
devices:
- driver: nvidia
count: all
capabilities: [gpu]
restart: unless-stopped