Rename setup_runtime.sh to setup.sh

setup.sh

@@ -0,0 +1,33 @@
+#!/bin/bash
+set -e
+
+echo "🔧 Starting environment setup..."
+
+# Miniconda install path
+CONDA_DIR=/opt/conda
+
+# Download and install Miniconda to /opt/conda
+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
+export PATH="$CONDA_DIR/bin:$PATH"
+
+# Create environment
+conda update -y -n base -c defaults conda
+conda env create -f /cosmos-predict1.yaml
+source $CONDA_DIR/bin/activate cosmos-predict1
+
+# Install Python dependencies
+pip install --no-cache-dir -r /requirements.txt
+pip install transformer-engine[pytorch]==1.12.0
+pip install git+https://github.com/microsoft/MoGe.git
+
+# Build and install Apex
+git clone https://github.com/NVIDIA/apex.git
+cd apex
+CUDA_HOME=$CONDA_DIR 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" .
+cd ..
+rm -rf apex
+
+echo "✅ Environment setup complete."

setup_runtime.sh

@@ -1,24 +0,0 @@
-#!/bin/bash
-
-echo "🚀 Installing Python packages..."
-
-# Install core dependencies from pip
-pip install --upgrade pip
-pip install --no-cache-dir -r /requirements.txt
-
-# Transformer Engine
-pip install transformer-engine[pytorch]==1.12.0
-
-# Apex
-git clone https://github.com/NVIDIA/apex
-cd apex
-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
-
-# Launch your app
-cd /app
-python app.py