setup.sh

#!/bin/bash

# Create conda environment

# Activate conda environment
CONDA_BASE=$(which conda)
eval "$($CONDA_BASE shell.bash hook)"
conda create --name korokoro -y python=3.10
conda activate korokoro

# Install cuda dependencies
conda install -c "nvidia/label/cuda-11.8.0" cuda-toolkit -y
pip install torch==2.1.2+cu118 torchvision==0.16.2+cu118 --extra-index-url https://download.pytorch.org/whl/cu118
conda install -c conda-forge colmap==3.8 -y


# Setup Grounding DINO
cd GroundingDINO/
mkdir weights
cd weights
wget -q https://github.com/IDEA-Research/GroundingDINO/releases/download/v0.1.0-alpha/groundingdino_swint_ogc.pth
cd ../..

# Install additional requirements
pip install -r requirements.txt

gdown "https://drive.google.com/u/1/uc?id=1-7x7qQfB7bIw2zV4Lr6-yhvMpjXC84Q5&confirm=t" 
pip install tinycudann-1.7-cp310-cp310-linux_x86_64.whl 

# Create folder to store result .obj files
mkdir results