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

RUN apt-get update && \
    apt-get install -y --no-install-recommends \
        python3.10 \
        python3-pip \
        openmpi-bin \
        libopenmpi-dev \
        python3-dev \
        build-essential && \
    rm -rf /var/lib/apt/lists/*

COPY ./requirements.txt /requirements.txt

RUN pip3 install -r requirements.txt

EXPOSE 80
EXPOSE 8080 

COPY ./ ./
CMD ["python3", "app.py"]