FROM nvidia/cuda:11.7.0-base-ubuntu22.04

RUN curl -sL "http://keyserver.ubuntu.com/pks/lookup?op=get&search=0xA4B469963BF863CC" | apt-key add
RUN curl -sL "http://keyserver.ubuntu.com/pks/lookup?op=get&search=0x871920D1991BC93C" | apt-key add
RUN apt-get update -y && apt-get install -y gcc python3-dev python3-venv python3-pip


COPY requirements.txt requirements.txt
RUN python3 -m pip install -r requirements.txt && rm requirements.txt

WORKDIR /home/worker

COPY src src
COPY entrypoint.py entrypoint.py
COPY worker.py worker.py
COPY config.ini config.ini

ENTRYPOINT [ "python3", "entrypoint.py"]