setup_runtime.sh

#!/bin/bash

echo "🚀 Starting runtime setup..."

# Set a writable conda install location
export CONDA_DIR="/tmp/miniconda3"
export PATH="$CONDA_DIR/bin:$PATH"

# Download and install Miniconda
mkdir -p $CONDA_DIR
wget https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-x86_64.sh -O /tmp/miniconda.sh
bash /tmp/miniconda.sh -b -u -p $CONDA_DIR
rm /tmp/miniconda.sh

# Activate conda (no source in non-interactive shell)
. "$CONDA_DIR/etc/profile.d/conda.sh"
conda activate base

# Create and activate the custom environment
conda env create --file /cosmos-predict1.yaml
conda activate cosmos-predict1

# Pip install
pip install --no-cache-dir -r /requirements.txt

# CUDA symlinks (optional, safe to fail)
ln -sf $CONDA_PREFIX/lib/python3.10/site-packages/nvidia/*/include/* $CONDA_PREFIX/include/ || true
ln -sf $CONDA_PREFIX/lib/python3.10/site-packages/nvidia/*/include/* $CONDA_PREFIX/include/python3.10 || true
ln -sf $CONDA_PREFIX/lib/python3.10/site-packages/triton/backends/nvidia/include/* $CONDA_PREFIX/include/ || true

# Transformer Engine
pip install transformer-engine[pytorch]==1.12.0

# Apex compilation
git clone https://github.com/NVIDIA/apex
cd apex
CUDA_HOME=$CONDA_PREFIX pip install -v --no-cache-dir --disable-pip-version-check --no-build-isolation \
  --config-settings "--build-option=--cpp_ext" --config-settings "--build-option=--cuda_ext" .

# MoGe
pip install git+https://github.com/microsoft/MoGe.git

# Return to app
cd /app
python app.py