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"