#!/usr/bin/env bash
# Source this file before building or running FoundationPose.
FP_ROOT="$(cd -- "$(dirname -- "${BASH_SOURCE[0]}")/.." && pwd)"
unset PYTHONPATH LD_LIBRARY_PATH
export CUDA_HOME="$FP_ROOT/.cuda/usr/local/cuda-12.8"
export PATH="$FP_ROOT/.venv/bin:$CUDA_HOME/bin:$PATH"
export TORCH_CUDA_ARCH_LIST=8.6
export MAX_JOBS=2
export PIP_CACHE_DIR="$FP_ROOT/.cache/pip"
export TORCH_EXTENSIONS_DIR="$FP_ROOT/.cache/torch_extensions"
export XDG_CACHE_HOME="$FP_ROOT/.cache"
export MPLCONFIGDIR="$FP_ROOT/.cache/matplotlib"
export CPATH="$CUDA_HOME/include"
for fp_inc in /home/timessage/python_envs/unitree_rl_mjlab/lib/python3.11/site-packages/nvidia/*/include; do
    export CPATH="$CPATH:$fp_inc"
done
unset fp_inc