embodiedgen/install/install_basic.sh
Xinjie e82f02a9a5
feat(pipe): Release 3D scene generation pipeline. (#25)
Release 3D scene generation pipeline and tag as v0.1.2.

---------

Co-authored-by: xinjie.wang <xinjie.wang@gpu-4090-dev015.hogpu.cc>
2025-07-21 23:31:15 +08:00

36 lines
1.3 KiB
Bash

#!/bin/bash
set -e
SCRIPT_DIR=$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)
source "$SCRIPT_DIR/_utils.sh"
try_install "Installing flash-attn..." \
"pip install flash-attn==2.7.0.post2 --no-build-isolation" \
"flash-attn installation failed."
try_install "Installing requirements.txt..." \
"pip install -r requirements.txt --use-deprecated=legacy-resolver --default-timeout=60" \
"requirements installation failed."
try_install "Installing kolors..." \
"pip install kolors@git+https://github.com/HochCC/Kolors.git" \
"kolors installation failed."
try_install "Installing kaolin..." \
"pip install kaolin@git+https://github.com/NVIDIAGameWorks/kaolin.git@v0.16.0" \
"kaolin installation failed."
log_info "Installing diff-gaussian-rasterization..."
TMP_DIR="/tmp/mip-splatting"
rm -rf "$TMP_DIR"
git clone --recursive https://github.com/autonomousvision/mip-splatting.git "$TMP_DIR"
pip install "$TMP_DIR/submodules/diff-gaussian-rasterization"
rm -rf "$TMP_DIR"
try_install "Installing gsplat..." \
"pip install git+https://github.com/nerfstudio-project/gsplat.git@v1.5.3" \
"gsplat installation failed."
try_install "Installing EmbodiedGen..." \
"pip install triton==2.1.0 --no-deps && pip install -e ." \
"EmbodiedGen installation failed."