File size: 314 Bytes
0db6a31 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 |
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
|