Dockerfile.nvidia

FROM nvidia/cuda:12.1.0-devel-ubuntu22.04

# Set environment variables to avoid interactive prompts
ENV DEBIAN_FRONTEND=noninteractive
ENV PYTHONUNBUFFERED=1

# Install Python 3.10 and system dependencies
RUN rm -f /etc/apt/apt.conf.d/docker-clean && \
    apt-get update && apt-get install -y --no-install-recommends \
    python3.10 \
    python3.10-dev \
    python3-pip \
    git \
    wget \
    curl \
    build-essential \
    ca-certificates \
    && apt-get clean \
    && rm -rf /var/lib/apt/lists/*

# Set Python 3.10 as default
RUN update-alternatives --install /usr/bin/python3 python3 /usr/bin/python3.10 1 && \
    update-alternatives --install /usr/bin/python python /usr/bin/python3.10 1

# Disable pip progress bar to avoid threading issues
ENV PIP_PROGRESS_BAR=off
ENV PIP_NO_INPUT=1

# Upgrade pip
RUN python3 -m pip install --no-cache-dir --upgrade pip setuptools wheel

# Copy requirements file
COPY requirements.txt .

# Install Python dependencies from requirements.txt (includes PyTorch)
RUN pip3 install --no-cache-dir -r requirements.txt

# Set working directory
WORKDIR /workspace/ai4c

# Copy project files (including .git directory to maintain git functionality)
COPY . .

# Reset any uncommitted changes to have a clean git state
RUN git reset --hard HEAD

# Set PYTHONPATH
ENV PYTHONPATH=/workspace/ai4c:$PYTHONPATH

# Set environment variables for AI4C (these should be overridden at runtime)
ENV AI4C_BASE_URL=""
ENV AI4C_API_KEY=""
ENV AI4C_API_MODEL_NAME=""

# Default command
CMD ["/bin/bash"]