Create setup_runtime.sh

setup_runtime.sh

@@ -0,0 +1,38 @@
+#!/bin/bash
+
+echo "🚀 Starting runtime setup..."
+
+# Install Miniconda
+mkdir -p ~/miniconda3
+wget https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-x86_64.sh -O ~/miniconda.sh
+bash ~/miniconda.sh -b -u -p ~/miniconda3
+rm ~/miniconda.sh
+source ~/miniconda3/bin/activate
+
+# Create and activate conda env
+~/miniconda3/bin/conda env create --file /cosmos-predict1.yaml
+source ~/miniconda3/bin/activate cosmos-predict1
+
+# Install requirements
+pip install --no-cache-dir -r /requirements.txt
+
+# CUDA header links (optional, safe to fail silently)
+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 --disable-pip-version-check --no-cache-dir --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
+
+# Launch Gradio app
+cd /app
+python app.py