26 lines
920 B
Bash
26 lines
920 B
Bash
#!/usr/bin/env bash
|
|
set -euo pipefail
|
|
|
|
NETWORK=${NETWORK:-unis-ai}
|
|
API_NAME=${API_NAME:-mineru-api}
|
|
WEB_NAME=${WEB_NAME:-mindmap-web}
|
|
WEB_PORT=${WEB_PORT:-9898}
|
|
MINERU_API_URL=${MINERU_API_URL:-http://${API_NAME}:8000}
|
|
MINERU_GPU_MEMORY_UTILIZATION=${MINERU_GPU_MEMORY_UTILIZATION:-0.5}
|
|
|
|
sudo docker network create "$NETWORK" >/dev/null 2>&1 || true
|
|
sudo docker rm -f "$API_NAME" "$WEB_NAME" >/dev/null 2>&1 || true
|
|
|
|
sudo docker run -d --name "$API_NAME" --network "$NETWORK" \
|
|
--privileged --ipc=host --shm-size=32g \
|
|
-e MINERU_MODEL_SOURCE=local \
|
|
-e PYTHONPATH=/app:/usr/local/corex-4.4.0/lib64/python3/dist-packages \
|
|
-e MINERU_GPU_MEMORY_UTILIZATION="$MINERU_GPU_MEMORY_UTILIZATION" \
|
|
unis-mindmap-api:latest \
|
|
mineru-api --host 0.0.0.0 --port 8000
|
|
|
|
sudo docker run -d --name "$WEB_NAME" --network "$NETWORK" \
|
|
-p "${WEB_PORT}:9898" \
|
|
-e MINERU_API_URL="$MINERU_API_URL" \
|
|
unis-mindmap-web:latest
|