Update setup_runtime.sh

setup_runtime.sh

@@ -2,37 +2,44 @@
 
 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
+# 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 header links (optional, safe to fail silently)
+# 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
+# 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 \
+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
 
-# Launch Gradio app
+# Return to app
 cd /app
 python app.py