Dockerfile.orchestrator

FROM python:3.10.13-slim

WORKDIR /app

ENV DEBIAN_FRONTEND=noninteractive
ENV TZ=Europe/London
RUN ln -snf /usr/share/zoneinfo/$TZ /etc/localtime && echo $TZ > /etc/timezone

RUN apt-get update && apt-get install -y \
    wget \ 
    git \
    curl \
    lsof \
    python3-dev \
    build-essential \
    python3-pip \
    apt-utils \
    vim \
    sudo \
    ffmpeg \
    libsm6 \
    libxext6 \
    python3-tk \
    python3-dev \
    git-lfs \
    unzip \
    && apt-get clean \
    && rm -rf /var/lib/apt/lists/*

RUN pip install --upgrade pip

COPY validator_orchestrator/requirements.txt /app/validator_orchestrator_requirements.txt

RUN pip install --no-cache-dir -r /app/validator_orchestrator_requirements.txt

# Install Docker
RUN apt-get update && apt-get install -y \
    ca-certificates \
    curl \
    gnupg \
    lsb-release \
    && mkdir -p /etc/apt/keyrings \
    && curl -fsSL https://download.docker.com/linux/debian/gpg | gpg --dearmor -o /etc/apt/keyrings/docker.gpg \
    && echo "deb [arch=$(dpkg --print-architecture) signed-by=/etc/apt/keyrings/docker.gpg] https://download.docker.com/linux/debian $(lsb_release -cs) stable" | tee /etc/apt/sources.list.d/docker.list > /dev/null \
    && apt-get update \
    && apt-get install -y docker-ce docker-ce-cli containerd.io docker-compose-plugin \
    && rm -rf /var/lib/apt/lists/*

# Install NVIDIA Container Toolkit
RUN curl -fsSL https://nvidia.github.io/libnvidia-container/gpgkey | gpg --dearmor -o /usr/share/keyrings/nvidia-container-toolkit-keyring.gpg \
    && curl -s -L https://nvidia.github.io/libnvidia-container/stable/deb/nvidia-container-toolkit.list | \
       sed 's#deb https://#deb [signed-by=/usr/share/keyrings/nvidia-container-toolkit-keyring.gpg] https://#g' | \
       tee /etc/apt/sources.list.d/nvidia-container-toolkit.list \
    && sed -i -e '/experimental/ s/^#//g' /etc/apt/sources.list.d/nvidia-container-toolkit.list \
    && apt-get update \
    && apt-get install -y nvidia-container-toolkit \
    && rm -rf /var/lib/apt/lists/*

# Configure Docker to use the NVIDIA runtime
RUN echo '{ \
  "default-runtime": "nvidia", \
  "runtimes": { \
    "nvidia": { \
      "path": "nvidia-container-runtime", \
      "runtimeArgs": [] \
    } \
  } \
}' | tee /etc/docker/daemon.json

COPY validator_orchestrator /app/validator_orchestrator

RUN chmod +x /app/validator_orchestrator/entrypoint.sh

USER root

CMD ["sh", "-c", "cd /app/validator_orchestrator && ./entrypoint.sh"]