FROM nvcr.io/nvidia/pytorch:23.11-py3

RUN pip install torchmetrics

ADD . /workspace

WORKDIR /workspace

CMD ["sh","run_eval.sh"]