From 2b5125b0c4801fe42210ab5920941c244a3014e3 Mon Sep 17 00:00:00 2001 From: Tyler Osterberg Date: Thu, 30 May 2024 13:03:50 -0700 Subject: [PATCH] [0.28.0-dlc][dlc] Update tag versions for containers (#1998) --- serving/docker/Dockerfile | 4 ++-- serving/docker/aarch64.Dockerfile | 4 ++-- serving/docker/pytorch-gpu.Dockerfile | 4 ++-- 3 files changed, 6 insertions(+), 6 deletions(-) diff --git a/serving/docker/Dockerfile b/serving/docker/Dockerfile index e96c885e7..ca0e70859 100644 --- a/serving/docker/Dockerfile +++ b/serving/docker/Dockerfile @@ -10,7 +10,7 @@ # BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, express or implied. See the License for # the specific language governing permissions and limitations under the License. FROM ubuntu:22.04 AS base -ARG djl_version=0.28.0~SNAPSHOT +ARG djl_version=0.28.0 COPY scripts scripts/ RUN mkdir -p /opt/djl/conf && \ @@ -75,5 +75,5 @@ RUN scripts/install_python.sh && \ apt-get clean -y && rm -rf /var/lib/apt/lists/* \ LABEL com.amazonaws.ml.engines.sagemaker.dlc.framework.djl.cpu-full="true" -LABEL com.amazonaws.ml.engines.sagemaker.dlc.framework.djl.v0-27-0.cpu-full="true" +LABEL com.amazonaws.ml.engines.sagemaker.dlc.framework.djl.v0-28-0.cpu-full="true" LABEL torch-version=$torch_version diff --git a/serving/docker/aarch64.Dockerfile b/serving/docker/aarch64.Dockerfile index eedaa73eb..7a0baea6c 100644 --- a/serving/docker/aarch64.Dockerfile +++ b/serving/docker/aarch64.Dockerfile @@ -10,7 +10,7 @@ # BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, express or implied. See the License for # the specific language governing permissions and limitations under the License. FROM arm64v8/ubuntu:22.04 -ARG djl_version=0.28.0~SNAPSHOT +ARG djl_version=0.28.0 ARG torch_version=2.2.2 EXPOSE 8080 @@ -57,7 +57,7 @@ RUN scripts/install_djl_serving.sh $djl_version && \ LABEL maintainer="djl-dev@amazon.com" LABEL dlc_major_version="1" LABEL com.amazonaws.ml.engines.sagemaker.dlc.framework.djl.aarch64="true" -LABEL com.amazonaws.ml.engines.sagemaker.dlc.framework.djl.v0-27-0.aarch64="true" +LABEL com.amazonaws.ml.engines.sagemaker.dlc.framework.djl.v0-28-0.aarch64="true" LABEL com.amazonaws.sagemaker.capabilities.multi-models="true" LABEL com.amazonaws.sagemaker.capabilities.accept-bind-to-port="true" LABEL djl-version=$djl_version diff --git a/serving/docker/pytorch-gpu.Dockerfile b/serving/docker/pytorch-gpu.Dockerfile index 9d548ba84..7e7e8880d 100644 --- a/serving/docker/pytorch-gpu.Dockerfile +++ b/serving/docker/pytorch-gpu.Dockerfile @@ -13,7 +13,7 @@ ARG version=12.1.1-cudnn8-devel-ubuntu22.04 FROM nvidia/cuda:$version as base -ARG djl_version=0.28.0~SNAPSHOT +ARG djl_version=0.28.0 ARG cuda_version=cu121 ARG torch_version=2.2.2 ARG torch_vision_version=0.17.2 @@ -73,7 +73,7 @@ CMD ["serve"] LABEL maintainer="djl-dev@amazon.com" LABEL dlc_major_version="1" LABEL com.amazonaws.ml.engines.sagemaker.dlc.framework.djl.pytorch-gpu="true" -LABEL com.amazonaws.ml.engines.sagemaker.dlc.framework.djl.v0-27-0.pytorch-cu121="true" +LABEL com.amazonaws.ml.engines.sagemaker.dlc.framework.djl.v0-28-0.pytorch-cu121="true" LABEL com.amazonaws.sagemaker.capabilities.multi-models="true" LABEL com.amazonaws.sagemaker.capabilities.accept-bind-to-port="true" LABEL djl-version=$djl_version