vkgs-sim / docker-compose.yml
kimjunhan's picture
Add files using upload-large-folder tool
5170de3 verified
Raw History Blame Contribute Delete
1.97 kB
# vkgs-sim ๊ฐœ๋ฐœ ์ปจํ…Œ์ด๋„ˆ (docker/README.md).
# docker compose build dev
# docker compose run --rm dev bash # ์…ธ
# docker compose run --rm dev vkgs-doctor # ์ ๊ฒ€
services:
dev:
build:
context: .
dockerfile: docker/Dockerfile
args:
CUDA_ARCH: ${CUDA_ARCH:-8.6}
UID: ${HOST_UID:-1000}
GID: ${HOST_GID:-1000}
image: ${IMAGE_NAME:-dzqf0000/vulkansim}:${IMAGE_TAG:-1.0}
# nvidia-container-toolkit ํ•„์š”. 'gpus: all' ๋Œ€์‹  ์˜ค๋ž˜๋œ compose ๋„ ์ฝ๋Š” ํ˜•ํƒœ.
deploy:
resources:
reservations:
devices:
- driver: nvidia
count: all
capabilities: [gpu]
runtime: nvidia
environment:
NVIDIA_VISIBLE_DEVICES: all
NVIDIA_DRIVER_CAPABILITIES: compute,utility,graphics,display
DISPLAY: ${DISPLAY:-:0}
VKGS_WORKSPACE: ${PWD}
NUSCENES_ROOT: ${NUSCENES_ROOT:-/home/mm/nas2/nuscenes_original_junhan/nuscenes/original/full}
volumes:
# ์ €์žฅ์†Œ๋ฅผ ํ˜ธ์ŠคํŠธ์™€ ๊ฐ™์€ ๊ฒฝ๋กœ๋กœ โ€” outputs/ ์˜ segments.txt ๋“ฑ์ด ์ ˆ๋Œ€ ๊ฒฝ๋กœ๋ฅผ ๋‹ด๋Š”๋‹ค.
- ${PWD}:${PWD}:rw
# C++ ๋นŒ๋“œ๋Š” ์ปจํ…Œ์ด๋„ˆ ์ „์šฉ ๋””๋ ‰ํ„ฐ๋ฆฌ๋กœ ๊ฐ€๋ฆฐ๋‹ค. ํ˜ธ์ŠคํŠธ build/ ๋Š” ํ˜ธ์ŠคํŠธ conda ์ปดํŒŒ์ผ๋Ÿฌ ๊ฒฝ๋กœ๋ฅผ ๋‹ด๊ณ  ์žˆ์–ด
# ๊ฐ™์ด ์“ฐ๋ฉด ์„œ๋กœ์˜ CMake ์บ์‹œ๋ฅผ ๊นฌ๋‹ค. nusim ์€ <์ €์žฅ์†Œ>/build/vkgs_viewer ๋ฅผ ์ฐพ์œผ๋ฏ€๋กœ ๊ฒฝ๋กœ๋Š” ๊ทธ๋Œ€๋กœ.
- ${PWD}/build-docker:${PWD}/build:rw
# nuScenes ์›๋ณธ์€ ๋ฐ˜๋“œ์‹œ ์ฝ๊ธฐ ์ „์šฉ (67 TB ๊ณต์šฉ NFS). ๊ฒฝ๋กœ๋„ ํ˜ธ์ŠคํŠธ์™€ ๊ฐ™๊ฒŒ.
- ${NUSCENES_ROOT:-/home/mm/nas2/nuscenes_original_junhan/nuscenes/original/full}:${NUSCENES_ROOT:-/home/mm/nas2/nuscenes_original_junhan/nuscenes/original/full}:ro
# X11 (์ฐฝ ๋ทฐ์–ด). ํ˜ธ์ŠคํŠธ์—์„œ ํ•œ ๋ฒˆ: xhost +local:
- /tmp/.X11-unix:/tmp/.X11-unix:rw
working_dir: ${PWD}
shm_size: "16gb"
ipc: host
network_mode: host
stdin_open: true
tty: true