File size: 1,831 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 36 | # 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
|