Ajouter start.sh pour lancer la carte en une commande, GPU facultatif

Co-Authored-By: Claude Opus 5.5 <noreply@anthropic.com>
This commit is contained in:
Antoine Jacquin
2026-09-27 21:02:19 +02:00
parent bae16923fc
commit 034dd5d827
4 changed files with 95 additions and 0 deletions

68
start.sh Executable file
View File

@ -0,0 +1,68 @@
#!/usr/bin/env bash
# Lancement local de la carte LiDAR en une commande (GPU facultatif).
#
# ./start.sh construit l'image et démarre la carte (port 8973)
# ./start.sh stop arrête la carte
# ./start.sh logs suit le journal du serveur
# ./start.sh process [opts] traitement ponctuel des dalles de input/
# (options passées au pipeline, ex. --file NOM)
#
# Sans GPU NVIDIA utilisable par Docker, docker-compose.cpu.yml est ajouté
# automatiquement (le pipeline tourne alors sur le CPU, plus lentement).
set -euo pipefail
cd "$(dirname "$0")"
# input/ et output/ créés au nom de l'utilisateur : créés par Docker, ils
# appartiendraient à root et le conteneur (uid 1000) ne pourrait pas y écrire.
mkdir -p input output
for d in input output; do
if [ ! -w "$d" ]; then
echo "Erreur : $d/ n'est pas accessible en écriture (propriétaire : $(stat -c %U "$d"))." >&2
echo "Corrigez avec : sudo chown -R $(id -u):$(id -g) $d" >&2
exit 1
fi
done
if [ "$(id -u)" != "1000" ]; then
echo "Attention : les conteneurs tournent sous l'uid 1000, vous êtes $(id -u) :" >&2
echo "input/ et output/ doivent rester accessibles en écriture à l'uid 1000." >&2
fi
# GPU : nvidia-smi présent ET Docker capable de le donner à un conteneur.
compose=(docker compose -f docker-compose.yml)
if command -v nvidia-smi >/dev/null 2>&1 && nvidia-smi >/dev/null 2>&1 \
&& docker run --rm --gpus all --entrypoint true busybox >/dev/null 2>&1; then
echo "GPU NVIDIA détecté : traitement accéléré."
else
echo "Pas de GPU utilisable : traitement sur le CPU (plus lent)."
compose+=(-f docker-compose.cpu.yml)
fi
case "${1:-up}" in
up)
"${compose[@]}" up -d --build serve
echo "Carte : http://localhost:8973/"
echo "Journal : ./start.sh logs — arrêt : ./start.sh stop"
;;
stop)
"${compose[@]}" down
;;
logs)
"${compose[@]}" logs -f serve
;;
process)
shift
if [ $# -gt 0 ]; then
# Des options remplacent la commande par défaut du service : la
# reprendre en entier (entrée, sortie, 0,2 m, tous les GPU).
"${compose[@]}" run --rm --build process \
python3 -m lidar_pipeline /data/input -o /data/output -r 0.2 -g all "$@"
else
"${compose[@]}" run --rm --build process
fi
;;
*)
sed -n '2,10p' "$0" | sed 's/^# \{0,1\}//'
exit 1
;;
esac