| FROM nvidia/cuda:11.6.2-base-ubuntu20.04 | |
| RUN apt update | |
| RUN apt upgrade -y | |
| RUN apt install -y python3 python3-pip | |
| RUN pip install torch torchvision torchaudio nnunet matplotlib | |
| ENV RESULTS_FOLDER="/root/results" | |
| COPY inference.sh inference.sh | |
| COPY nnUNet_trained_models /root/results | |
| RUN chmod +x inference.sh | |