|
a |
|
b/Dockerfile |
|
|
1 |
FROM nvidia/cuda:11.0-base |
|
|
2 |
RUN apt-get update && apt install -y software-properties-common |
|
|
3 |
RUN add-apt-repository ppa:deadsnakes/ppa |
|
|
4 |
RUN apt-get update |
|
|
5 |
RUN apt install -y python3.8 |
|
|
6 |
RUN apt install -y python3-pip |
|
|
7 |
RUN apt install -y git |
|
|
8 |
COPY requirements_v2.txt /usr/local/bin |
|
|
9 |
RUN pip3 install -r /usr/local/bin/requirements_v2.txt |
|
|
10 |
RUN pip3 install torch==1.9.1+cu111 torchvision==0.10.1+cu111 torchaudio==0.9.1 -f https://download.pytorch.org/whl/torch_stable.html |
|
|
11 |
COPY nnUNet/ /usr/local/bin/nnUNet/ |
|
|
12 |
COPY trained_models /usr/local/bin/trained_models/ |
|
|
13 |
RUN pip3 install -U setuptools |
|
|
14 |
RUN pip3 install -e /usr/local/bin/nnUNet |
|
|
15 |
ENV RESULTS_FOLDER=/usr/local/bin/trained_models/ |
|
|
16 |
COPY inference_v2.py /usr/local/bin |
|
|
17 |
|
|
|
18 |
ENTRYPOINT ["python3", "/usr/local/bin/inference_v2.py"] |