Update setup_runtime.sh

setup_runtime.sh

@@ -1,45 +1,24 @@
 #!/bin/bash
 
-echo "🚀 Starting runtime setup..."
+echo "🚀 Installing Python packages..."
 
-# 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
+# Install core dependencies from pip
+pip install --upgrade pip
 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
+# Apex
 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 \
+CUDA_HOME=/usr/local/cuda 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" .
+cd ..
 
 # MoGe
 pip install git+https://github.com/microsoft/MoGe.git
 
-# Return to app
+# Launch your app
 cd /app
 python app.py