From c5c3ace725b9cd8fbd9d75a77ef930581514a94a Mon Sep 17 00:00:00 2001 From: Joschuan Santana Date: Wed, 15 May 2024 15:21:20 -0300 Subject: [PATCH 1/2] Implement sagemaker inference dockerfile images --- src/ecr/roof-energy-inference-cpu/Dockerfile | 18 ++++++++++++++++++ src/ecr/roof-energy-inference-gpu/Dockerfile | 18 ++++++++++++++++++ 2 files changed, 36 insertions(+) create mode 100644 src/ecr/roof-energy-inference-cpu/Dockerfile create mode 100644 src/ecr/roof-energy-inference-gpu/Dockerfile diff --git a/src/ecr/roof-energy-inference-cpu/Dockerfile b/src/ecr/roof-energy-inference-cpu/Dockerfile new file mode 100644 index 0000000..0fcab28 --- /dev/null +++ b/src/ecr/roof-energy-inference-cpu/Dockerfile @@ -0,0 +1,18 @@ +# https://aws.amazon.com/releasenotes/available-deep-learning-containers-images/ +# https://github.com/aws/deep-learning-containers/blob/master/available_images.md + +FROM 763104351884.dkr.ecr.us-east-1.amazonaws.com/pytorch-training:2.1.0-cpu-py310-ubuntu20.04-sagemaker + +# RUN adduser --system --group nonroot + +RUN apt update \ + && apt install --no-install-recommends -y binutils libproj-dev gdal-bin libgdal-dev \ + && apt clean + +RUN python3 -m pip install --upgrade pip --no-cache-dir \ + && pip3 install --upgrade --no-cache-dir astral==3.2 groundingdino-py==0.4.0 leafmap==0.31.3 openpyxl==3.1.2 segment-geospatial==0.10.2 + +RUN pip3 install setuptools==57.5.0 \ + && pip3 install gdal==$(gdal-config --version) + +# USER nonroot diff --git a/src/ecr/roof-energy-inference-gpu/Dockerfile b/src/ecr/roof-energy-inference-gpu/Dockerfile new file mode 100644 index 0000000..426c412 --- /dev/null +++ b/src/ecr/roof-energy-inference-gpu/Dockerfile @@ -0,0 +1,18 @@ +# https://docs.aws.amazon.com/deep-learning-containers/latest/devguide/deep-learning-containers-images.html +# https://github.com/aws/deep-learning-containers/blob/master/available_images.md + +FROM 763104351884.dkr.ecr.us-east-1.amazonaws.com/pytorch-training:2.1.0-gpu-py310-cu121-ubuntu20.04-sagemaker + +# RUN adduser --system --group nonroot + +RUN apt update \ + && apt install --no-install-recommends -y binutils libproj-dev gdal-bin libgdal-dev \ + && apt clean + +RUN python3 -m pip install --upgrade pip --no-cache-dir \ + && pip3 install --upgrade --no-cache-dir astral==3.2 groundingdino-py==0.4.0 leafmap==0.31.3 openpyxl==3.1.2 segment-geospatial==0.10.2 + +RUN pip3 install setuptools==57.5.0 \ + && pip3 install gdal==$(gdal-config --version) + +# USER nonroot From 6be7a0920968e5bca915941aaa2c16bc73c4b8f1 Mon Sep 17 00:00:00 2001 From: Joschuan Santana Date: Wed, 15 May 2024 15:24:24 -0300 Subject: [PATCH 2/2] Minor update --- src/ecr/roof-energy-inference-cpu/Dockerfile | 6 +----- src/ecr/roof-energy-inference-gpu/Dockerfile | 4 ---- 2 files changed, 1 insertion(+), 9 deletions(-) diff --git a/src/ecr/roof-energy-inference-cpu/Dockerfile b/src/ecr/roof-energy-inference-cpu/Dockerfile index 0fcab28..7139166 100644 --- a/src/ecr/roof-energy-inference-cpu/Dockerfile +++ b/src/ecr/roof-energy-inference-cpu/Dockerfile @@ -1,10 +1,8 @@ -# https://aws.amazon.com/releasenotes/available-deep-learning-containers-images/ +# https://docs.aws.amazon.com/deep-learning-containers/latest/devguide/deep-learning-containers-images.html # https://github.com/aws/deep-learning-containers/blob/master/available_images.md FROM 763104351884.dkr.ecr.us-east-1.amazonaws.com/pytorch-training:2.1.0-cpu-py310-ubuntu20.04-sagemaker -# RUN adduser --system --group nonroot - RUN apt update \ && apt install --no-install-recommends -y binutils libproj-dev gdal-bin libgdal-dev \ && apt clean @@ -14,5 +12,3 @@ RUN python3 -m pip install --upgrade pip --no-cache-dir \ RUN pip3 install setuptools==57.5.0 \ && pip3 install gdal==$(gdal-config --version) - -# USER nonroot diff --git a/src/ecr/roof-energy-inference-gpu/Dockerfile b/src/ecr/roof-energy-inference-gpu/Dockerfile index 426c412..6b34ddc 100644 --- a/src/ecr/roof-energy-inference-gpu/Dockerfile +++ b/src/ecr/roof-energy-inference-gpu/Dockerfile @@ -3,8 +3,6 @@ FROM 763104351884.dkr.ecr.us-east-1.amazonaws.com/pytorch-training:2.1.0-gpu-py310-cu121-ubuntu20.04-sagemaker -# RUN adduser --system --group nonroot - RUN apt update \ && apt install --no-install-recommends -y binutils libproj-dev gdal-bin libgdal-dev \ && apt clean @@ -14,5 +12,3 @@ RUN python3 -m pip install --upgrade pip --no-cache-dir \ RUN pip3 install setuptools==57.5.0 \ && pip3 install gdal==$(gdal-config --version) - -# USER nonroot