FROM nvidia/cuda:12.9.0-devel-ubuntu24.04

RUN apt-get update && apt-get install -y \
    bash \
    build-essential \
    ninja-build \
    python3 \
    python3-pip \
    && rm -rf /var/lib/apt/lists/*

ENV PATH="/usr/local/cuda/bin:${PATH}"

RUN pip3 install --break-system-packages numpy

WORKDIR /workspace
RUN mkdir -p /workspace/compute_eval_work
