EmbodiedGen/install.sh at master · HorizonRobotics/EmbodiedGen · GitHub

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
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
#!/bin/bash
set -e

STAGE=$1 # "basic" | "scene3d" | "room" | "affordance" | "cu126" | "all"
STAGE=${STAGE:-basic}

REPO_ROOT=$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)
source "$REPO_ROOT/install/_utils.sh"
cd "$REPO_ROOT"

case "$STAGE" in
    basic|scene3d|room|affordance|cu126|all) ;;
    *)
        log_error "Unknown installation stage: $STAGE"
        log_error "Usage: bash install.sh [basic|scene3d|room|affordance|cu126|all]"
        exit 1
        ;;
esac

git config http.postBuffer 524288000

log_info "===== Starting installation stage: $STAGE ====="

if [[ "$STAGE" != "cu126" ]]; then
    bash "$REPO_ROOT/install/init_submodules.sh" "$STAGE"
fi

if [[ "$STAGE" == "basic" || "$STAGE" == "all" ]]; then
    bash "$REPO_ROOT/install/install_basic.sh"
fi

if [[ "$STAGE" == "scene3d" || "$STAGE" == "all" ]]; then
    PANO2ROOM_PATH="$REPO_ROOT/thirdparty/pano2room"
    if [ -d "$PANO2ROOM_PATH" ]; then
        printf "__pycache__/\n.gitignore\n" > "$PANO2ROOM_PATH/.gitignore"
        log_info "Added .gitignore to ignore __pycache__ in $PANO2ROOM_PATH"
    fi
    bash "$REPO_ROOT/install/install_scene3d.sh"
fi

if [[ "$STAGE" == "room" || "$STAGE" == "all" ]]; then
    bash "$REPO_ROOT/install/install_room.sh"
fi

if [[ "$STAGE" == "affordance" || "$STAGE" == "all" ]]; then
    bash "$REPO_ROOT/install/install_affordance.sh"
fi

if [[ "$STAGE" == "cu126" ]]; then
    bash "$REPO_ROOT/install/install_cu126.sh"
fi

# Global constraints for all stages
python -m pip install numpy==1.26.4

if [[ "$STAGE" != "cu126" ]]; then
    try_install "Refreshing EmbodiedGen editable install..." \
        "python -m pip install -e ." \
        "EmbodiedGen editable installation refresh failed."
fi

log_info "===== Installation completed successfully. ====="