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