File size: 1,370 Bytes
4c75661 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 | #!/usr/bin/env bash
# ๊ฐ๋ฐ ์ด๋ฏธ์ง ๋น๋. ์ต์ด 20~40๋ถ, ์ดํ ์บ์๊ฐ ๋๋ถ๋ถ์ ๋จน๋๋ค.
set -euo pipefail
. "$(dirname "${BASH_SOURCE[0]}")/_docker.sh"
say "๋น๋ ์ ์ ๊ฒ"
# nvidia ๋ฐํ์์ด ์์ผ๋ฉด ์ด๋ฏธ์ง๋ ๋น๋๋ผ๋ ์คํ ์์ ์ GPU ๋ฅผ ๋ชป ๋ณธ๋ค.
# ์ฌ๊ธฐ์ ๋ฏธ๋ฆฌ ์๋ ค ์ฃผ๋ ํธ์ด 40๋ถ ๋ค์ ์๋ ๊ฒ๋ณด๋ค ๋ซ๋ค.
if ! "${DOCKER[@]}" info 2>/dev/null | grep -qi 'nvidia'; then
warn "docker ๋ฐ๋ชฌ์ nvidia ๋ฐํ์์ด ์ ๋ณด์
๋๋ค."
warn "SETUP.md 2๋จ๊ณ(nvidia-container-toolkit)๋ฅผ ๋จผ์ ๋๋ด์ธ์. ๋น๋๋ ๊ณ์ํฉ๋๋ค."
else
ok "nvidia ๋ฐํ์ ๋ฑ๋ก๋จ"
fi
: "${NUSCENES_ROOT:?.env ์ NUSCENES_ROOT ๊ฐ ์์ต๋๋ค}"
if [[ -d "$NUSCENES_ROOT/v1.0-trainval" ]]; then
ok "nuScenes ๋ฉํ๋ฐ์ดํฐ ํ์ธ: $NUSCENES_ROOT/v1.0-trainval"
else
warn "$NUSCENES_ROOT/v1.0-trainval ์ด ์ ๋ณด์
๋๋ค (NFS ์๋ต์ด ๋๋ฆด ์ ์์)."
fi
AVAIL=$(df -BG --output=avail /var/lib/docker 2>/dev/null | tail -1 | tr -dc '0-9')
if [[ -n "${AVAIL:-}" && "$AVAIL" -lt 30 ]]; then
warn "/var/lib/docker ์ฌ์ ๊ณต๊ฐ ${AVAIL}GB. ์ด๋ฏธ์ง๊ฐ 12~16GB ์ด๋ฏ๋ก ๋ถ์กฑํ ์ ์์ต๋๋ค."
fi
say "์ด๋ฏธ์ง ๋น๋: ${IMAGE_NAME:-vkgs-sim}:${IMAGE_TAG:-dev} (CUDA_ARCH=${CUDA_ARCH:-8.6})"
export DOCKER_BUILDKIT=1
dc build "$@" dev
ok "๋น๋ ์๋ฃ"
echo " ๋ค์: bash scripts/run.sh"
|