Create model dir to prevent issues with no code experience in SageMaker

deepjavalibrary · Apr 28, 2023 · 8d8db5a · 8d8db5a
1 parent 471fec7
commit 8d8db5a
Show file tree

Hide file tree

Showing 7 changed files with 13 additions and 6 deletions.
diff --git a/serving/docker/Dockerfile b/serving/docker/Dockerfile
@@ -13,7 +13,8 @@ FROM ubuntu:20.04 AS base
 ARG djl_version=0.23.0~SNAPSHOT
 
 COPY scripts scripts/
-RUN mkdir -p /opt/djl/conf
+RUN mkdir -p /opt/djl/conf && \
+    mkdir -p /opt/ml/model
 COPY config.properties /opt/djl/conf/
 RUN scripts/install_djl_serving.sh $djl_version && \
     scripts/install_s5cmd.sh x64 && \

diff --git a/serving/docker/aarch64.Dockerfile b/serving/docker/aarch64.Dockerfile
@@ -30,7 +30,8 @@ CMD ["serve"]
 
 COPY scripts scripts/
 RUN mkdir -p /opt/djl/conf && \
-    mkdir -p /opt/djl/deps
+    mkdir -p /opt/djl/deps && \
+    mkdir -p /opt/ml/model
 COPY config.properties /opt/djl/conf/
 
 RUN scripts/install_djl_serving.sh $djl_version && \

diff --git a/serving/docker/deepspeed.Dockerfile b/serving/docker/deepspeed.Dockerfile
@@ -39,7 +39,8 @@ CMD ["serve"]
 COPY scripts scripts/
 RUN mkdir -p /opt/djl/conf && \
     mkdir -p /opt/djl/deps && \
-    mkdir -p /opt/djl/partition
+    mkdir -p /opt/djl/partition && \
+    mkdir -p /opt/ml/model
 COPY config.properties /opt/djl/conf/config.properties
 COPY partition /opt/djl/partition
 

diff --git a/serving/docker/fastertransformer.Dockerfile b/serving/docker/fastertransformer.Dockerfile
@@ -40,7 +40,8 @@ CMD ["serve"]
 COPY scripts scripts/
 RUN mkdir -p /opt/djl/conf && \
     mkdir -p /opt/djl/deps && \
-    mkdir -p /opt/djl/partition
+    mkdir -p /opt/djl/partition && \
+    mkdir -p /opt/ml/model
 COPY config.properties /opt/djl/conf/config.properties
 COPY partition /opt/djl/partition
 

diff --git a/serving/docker/pytorch-cu118.Dockerfile b/serving/docker/pytorch-cu118.Dockerfile
@@ -39,6 +39,7 @@ COPY scripts scripts/
 RUN chmod +x /usr/local/bin/dockerd-entrypoint.sh && \
     scripts/install_djl_serving.sh $djl_version && \
     mkdir -p /opt/djl/bin && cp scripts/telemetry.sh /opt/djl/bin && \
+    mkdir -p /opt/ml/model && \
     echo "${djl_version} pytorchgpu" > /opt/djl/bin/telemetry && \
     scripts/install_djl_serving.sh $djl_version ${torch_version} && \
     scripts/install_python.sh ${python_version} && \

diff --git a/serving/docker/pytorch-inf1.Dockerfile b/serving/docker/pytorch-inf1.Dockerfile
@@ -39,7 +39,8 @@ CMD ["serve"]
 
 COPY scripts scripts/
 RUN mkdir -p /opt/djl/conf && \
-    mkdir -p /opt/djl/deps
+    mkdir -p /opt/djl/deps && \
+    mkdir -p /opt/ml/model 
 COPY config.properties /opt/djl/conf/
 RUN scripts/install_djl_serving.sh $djl_version && \
     mkdir -p /opt/djl/bin && cp scripts/telemetry.sh /opt/djl/bin && \

diff --git a/serving/docker/pytorch-inf2.Dockerfile b/serving/docker/pytorch-inf2.Dockerfile
@@ -45,7 +45,8 @@ CMD ["serve"]
 
 COPY scripts scripts/
 RUN mkdir -p /opt/djl/conf && \
-    mkdir -p /opt/djl/deps
+    mkdir -p /opt/djl/deps && \
+    mkdir -p /opt/ml/model
 COPY config.properties /opt/djl/conf/
 RUN mkdir -p /opt/djl/bin && cp scripts/telemetry.sh /opt/djl/bin && \
     echo "${djl_version} inf2" > /opt/djl/bin/telemetry && \