PROJECT_DIR=$(pwd)

# Pytorch
pip3 install -i https://download.pytorch.org/whl/cu128 -U torch torchvision torchaudio
pip3 install -i https://download.pytorch.org/whl/cu128 -U xformers

# A modified gaussian splatting (+ alpha, depth, normal rendering)
cd extensions && git clone https://github.com/BaowenZ/RaDe-GS.git --recursive && cd RaDe-GS/submodules
pip3 install ./diff-gaussian-rasterization
cd ${PROJECT_DIR}

# Others
pip3 install -U gpustat
pip3 install -U -r settings/requirements.txt
sudo apt-get install -y ffmpeg
pip3 install -U -r settings/requirements-dev.txt