Skip to content

Commit

Permalink
[0.28.0-dlc][dlc] Update tag versions for containers (#1998)
Browse files Browse the repository at this point in the history
  • Loading branch information
tosterberg authored May 30, 2024
1 parent 443f952 commit 2b5125b
Show file tree
Hide file tree
Showing 3 changed files with 6 additions and 6 deletions.
4 changes: 2 additions & 2 deletions serving/docker/Dockerfile
Original file line number Diff line number Diff line change
Expand Up @@ -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 && \
Expand Down Expand Up @@ -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
4 changes: 2 additions & 2 deletions serving/docker/aarch64.Dockerfile
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down Expand Up @@ -57,7 +57,7 @@ RUN scripts/install_djl_serving.sh $djl_version && \
LABEL maintainer="[email protected]"
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
Expand Down
4 changes: 2 additions & 2 deletions serving/docker/pytorch-gpu.Dockerfile
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down Expand Up @@ -73,7 +73,7 @@ CMD ["serve"]
LABEL maintainer="[email protected]"
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
Expand Down

0 comments on commit 2b5125b

Please sign in to comment.