
FROM nvcr.io/nvidia/pytorch:23.01-py3
# FROM nvcr.io/nvidia/pytorch:23.05-py3

RUN pip install nnunetv2

ENV          nnUNet_raw="/nnUNet/raw"
ENV nnUNet_preprocessed="/nnUNet/preprocessed"
ENV      nnUNet_results="/nnUNet/results"

WORKDIR /work

COPY results /nnUNet/results
COPY predict.sh predict.sh

# ENTRYPOINT ["tail", "-f", "/dev/null"]
# ENTRYPOINT ["nvidia-smi"]
ENTRYPOINT /work/predict.sh
