setup_environment.sh

#!/bin/bash

# Install system dependencies
apt-get update && apt-get install -y \
    ffmpeg \
    git \
    wget \
    curl \
    build-essential \
    && rm -rf /var/lib/apt/lists/*

# Install PyTorch first
pip install --no-cache-dir torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu

# Install mmcv-full with pre-built wheel
pip install --no-cache-dir mmcv-full -f https://download.openmmlab.com/mmcv/dist/cpu/torch2.0/index.html

# Install other dependencies
pip install --no-cache-dir -r requirements.txt

echo "Environment setup completed!"