coder10/runpod-training repository overview

    FROM nvidia/cuda:12.1.1-cudnn8-runtime-ubuntu22.04

    # ------------------------------------------------------------
    # Environment setup
    # ------------------------------------------------------------
    ENV DEBIAN_FRONTEND=noninteractive \
        PIP_DISABLE_PIP_VERSION_CHECK=1 \
        PYTHONDONTWRITEBYTECODE=1 \
        PYTHONUNBUFFERED=1
    
    # ------------------------------------------------------------
    # System dependencies
    # ------------------------------------------------------------
    RUN apt-get update && apt-get install -y \
        python3 \
        python3-pip \
        python3-venv \
        git \
        wget \
        curl \
        ca-certificates \
        awscli \
        && rm -rf /var/lib/apt/lists/*
    
    # ------------------------------------------------------------
    # Python virtual environment
    # ------------------------------------------------------------
    RUN python3 -m venv /opt/venv
    ENV PATH="/opt/venv/bin:$PATH"
    
    RUN pip install --upgrade pip setuptools wheel
    
    # ------------------------------------------------------------
    # PyTorch (CUDA 12.1)
    # ------------------------------------------------------------
    RUN pip install --index-url https://download.pytorch.org/whl/cu121 \
        torch \
        torchvision \
        torchaudio
    
    # ------------------------------------------------------------
    # ML training stack (PINNED versions – very important)
    # ------------------------------------------------------------
    RUN pip install \
        accelerate==0.33.0 \
        transformers==4.44.2 \
        peft==0.12.0 \
        diffusers==0.30.2 \
        datasets==2.20.0 \
        safetensors==0.4.4 \
        pillow==10.4.0 \
        xformers==0.0.27.post2
    
    # ------------------------------------------------------------
    # Workspace
    # ------------------------------------------------------------
    WORKDIR /workspace
    
    # ------------------------------------------------------------
    # Default command
    # ------------------------------------------------------------
    CMD ["bash"]