Skip to content

Commit db06ff5

Browse files
michal2409nv-kkudrynski
michal2409
authored and
nv-kkudrynski
committed
[nnUNet/PyT] Update container to 21.11, use DynUNet from MONAI for model template, code refactor
1 parent a8c42a1 commit db06ff5

24 files changed

Lines changed: 1009 additions & 1068 deletions

File tree

‎PyTorch/Segmentation/nnUNet/Dockerfile‎

Lines changed: 5 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -1,11 +1,10 @@
1-
ARG FROM_IMAGE_NAME=nvcr.io/nvidia/pytorch:21.02-py3
1+
ARG FROM_IMAGE_NAME=nvcr.io/nvidia/pytorch:21.11-py3
22
FROM ${FROM_IMAGE_NAME}
33

4-
ADD ./requirements.txt /
5-
RUN pip install --disable-pip-version-check -r /requirements.txt
6-
RUN pip install monai==0.7.0 --no-dependencies
7-
RUN pip install numpy --upgrade
8-
RUN pip install --extra-index-url https://developer.download.nvidia.com/compute/redist/ nvidia-dali-cuda110==1.5.0
4+
RUN pip install nvidia-pyindex
5+
ADD ./requirements.txt .
6+
RUN pip install --disable-pip-version-check -r requirements.txt
7+
RUN pip install monai==0.8.0 --no-dependencies
98

109
RUN curl "https://awscli.amazonaws.com/awscli-exe-linux-x86_64.zip" -o "awscliv2.zip"
1110
RUN unzip -qq awscliv2.zip

‎PyTorch/Segmentation/nnUNet/Dockerfile-Triton‎

Lines changed: 7 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -1,18 +1,14 @@
1-
ARG FROM_IMAGE_NAME=nvcr.io/nvidia/pytorch:21.02-py3
1+
ARG FROM_IMAGE_NAME=nvcr.io/nvidia/pytorch:21.11-py3
22
FROM ${FROM_IMAGE_NAME}
33

4-
ADD ./triton/requirements.txt /
5-
RUN pip install --disable-pip-version-check -r /requirements.txt
4+
RUN pip install nvidia-pyindex
5+
ADD ./triton/requirements.txt .
6+
RUN pip install --disable-pip-version-check -r requirements.txt
67
RUN apt-get update && apt-get install -y libb64-dev libb64-0d
78

8-
ADD ./requirements.txt /
9-
RUN pip install --disable-pip-version-check -r /requirements.txt
10-
RUN pip install monai==0.7.0 --no-dependencies
11-
RUN pip install numpy --upgrade
12-
RUN pip install nvidia-pyindex==1.0.9
13-
RUN pip install nvidia-dlprof==1.2.0
14-
RUN pip install nvidia_dlprof_pytorch_nvtx==1.2.0
15-
RUN pip install --extra-index-url https://developer.download.nvidia.com/compute/redist/ nvidia-dali-cuda110==1.5.0
9+
ADD ./requirements.txt .
10+
RUN pip install --disable-pip-version-check -r requirements.txt
11+
RUN pip install monai==0.8.0 --no-dependencies
1612

1713
RUN curl "https://awscli.amazonaws.com/awscli-exe-linux-x86_64.zip" -o "awscliv2.zip"
1814
RUN unzip -qq awscliv2.zip

‎PyTorch/Segmentation/nnUNet/README.md‎

Lines changed: 88 additions & 87 deletions
Large diffs are not rendered by default.

0 commit comments

Comments
 (0)