vkgs-sim / scripts /build.sh
kimjunhan's picture
Add files using upload-large-folder tool
4c75661 verified
Raw History Blame Contribute Delete
1.37 kB
#!/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"