vkgs-sim / scripts /local_env.sh
kimjunhan's picture
Add files using upload-large-folder tool
4c75661 verified
Raw History Blame Contribute Delete
1.83 kB
# shellcheck shell=bash
# ๋กœ์ปฌ(Docker ์—†์Œ) ๊ฐœ๋ฐœ ํ™˜๊ฒฝ ํ™œ์„ฑํ™”. ์‚ฌ์šฉ๋ฒ•: source scripts/local_env.sh
#
# docker/Dockerfile ์˜ ENV ๋“ค๊ณผ docker-compose ๊ฐ€ ๋„ฃ์–ด ์ฃผ๋˜ ๊ฐ’์„ ์—ฌ๊ธฐ์„œ ๋Œ€์‹  ๊ฑด๋‹ค.
# ํ™˜๊ฒฝ ์ž์ฒด๋Š” scripts/local_setup.sh ๊ฐ€ ๋งŒ๋“ ๋‹ค.
_VKGS_ROOT="$(cd "$(dirname "${BASH_SOURCE[0]}")/.." && pwd)"
export MAMBA_ROOT_PREFIX="${MAMBA_ROOT_PREFIX:-$HOME/micromamba}"
eval "$("$HOME/.local/bin/micromamba" shell hook -s bash)"
micromamba activate vkgs
# --- Dockerfile ENV ๋Œ€์‘ --------------------------------------------------
export VKGS_WORKSPACE="$_VKGS_ROOT"
export CUDA_HOME="$CONDA_PREFIX" # conda ์˜ cuda-nvcc 12.1
export TORCH_CUDA_ARCH_LIST="8.6" # RTX 3060 Ti
# conda-forge ์˜ CUDA ๋Š” ํ—ค๋”/๋ผ์ด๋ธŒ๋Ÿฌ๋ฆฌ๋ฅผ targets/ ์•„๋ž˜์— ๋‘”๋‹ค. torch ์˜ ํ™•์žฅ ๋นŒ๋”๋Š”
# $CUDA_HOME/include ๋งŒ ๋ณด๋ฏ€๋กœ ์ด๊ฒŒ ์—†์œผ๋ฉด gsplat JIT ๊ฐ€ cuda_runtime_api.h ๋ฅผ ๋ชป ์ฐพ๋Š”๋‹ค.
_CUDA_TGT="$CONDA_PREFIX/targets/x86_64-linux"
export CPATH="$_CUDA_TGT/include${CPATH:+:$CPATH}"
export LIBRARY_PATH="$_CUDA_TGT/lib:$CONDA_PREFIX/lib${LIBRARY_PATH:+:$LIBRARY_PATH}"
unset _CUDA_TGT
export PYTHONUNBUFFERED=1
export CCACHE_DIR="$_VKGS_ROOT/cache/ccache"
# nuScenes: ํ˜ธ์ŠคํŠธ์—์„œ๋Š” NAS ์›๋ณธ์„ ์ง์ ‘ ๋ณธ๋‹ค (์ฝ๊ธฐ ์ „์šฉ โ€” ์ ˆ๋Œ€ ์“ฐ์ง€ ์•Š๋Š”๋‹ค).
export NUSCENES_ROOT="${NUSCENES_ROOT:-/home/mm/nas2/nuscenes_original_junhan/nuscenes/original/full}"
# --- Vulkan -------------------------------------------------------------
# conda ์˜ libvulkan-loader ๋Š” ์ž๊ธฐ prefix ์•„๋ž˜ icd.d ๋งŒ ๋’ค์ง„๋‹ค.
# ํ˜ธ์ŠคํŠธ NVIDIA ๋“œ๋ผ์ด๋ฒ„์˜ ICD ๋ฅผ ๋ช…์‹œํ•ด์•ผ GPU ๊ฐ€ ๋ณด์ธ๋‹ค (์•ˆ ํ•˜๋ฉด ์žฅ์น˜ 0๊ฐœ).
export VK_DRIVER_FILES=/usr/share/vulkan/icd.d/nvidia_icd.json
export VK_ICD_FILENAMES="$VK_DRIVER_FILES" # ๊ตฌ๋ฒ„์ „ ๋กœ๋”์šฉ ์ด๋ฆ„
unset _VKGS_ROOT