Update DLC Containers

Author: dk19y

test/container/1.6.0/Dockerfile.dlc.cpu

@@ -1,6 +1,23 @@
 ARG region
 FROM 763104351884.dkr.ecr.$region.amazonaws.com/pytorch-inference:1.5.0-cpu-py3
 
+ARG TS_VERSION=0.1.1
+RUN apt-get update \
+    && apt-get install -y --no-install-recommends software-properties-common \
+    && add-apt-repository ppa:openjdk-r/ppa \
+    && apt-get update \
+    && apt-get install -y --no-install-recommends openjdk-11-jdk
+
+RUN pip install torchserve==$TS_VERSION \
+    && pip install torch-model-archiver==$TS_VERSION
+
+RUN pip uninstall torch \
+    && pip uninstall torchvision \
+    && pip install torch=1.6.0 \
+    && pip install torchvision=0.7.0
+
 COPY dist/sagemaker_pytorch_inference-*.tar.gz /sagemaker_pytorch_inference.tar.gz
 RUN pip install --upgrade --no-cache-dir /sagemaker_pytorch_inference.tar.gz && \
     rm /sagemaker_pytorch_inference.tar.gz
+
+CMD ["torchserve", "--start", "--ts-config", "/home/model-server/config.properties", "--model-store", "/home/model-server/"]

test/container/1.6.0/Dockerfile.dlc.gpu

@@ -1,6 +1,23 @@
 ARG region
-FROM 763104351884.dkr.ecr.$region.amazonaws.com/pytorch-inference:1.5.0-gpu-py3
+FROM 763104351884.dkr.ecr.$region.amazonaws.com/pytorch-inference:1.5.0-cpu-py3
+
+ARG TS_VERSION=0.1.1
+RUN apt-get update \
+    && apt-get install -y --no-install-recommends software-properties-common \
+    && add-apt-repository ppa:openjdk-r/ppa \
+    && apt-get update \
+    && apt-get install -y --no-install-recommends openjdk-11-jdk
+
+RUN pip install torchserve==$TS_VERSION \
+    && pip install torch-model-archiver==$TS_VERSION
+
+RUN pip uninstall torch \
+    && pip uninstall torchvision \
+    && pip install torch=1.6.0 \
+    && pip install torchvision=0.7.0
 
 COPY dist/sagemaker_pytorch_inference-*.tar.gz /sagemaker_pytorch_inference.tar.gz
 RUN pip install --upgrade --no-cache-dir /sagemaker_pytorch_inference.tar.gz && \
     rm /sagemaker_pytorch_inference.tar.gz
+
+CMD ["torchserve", "--start", "--ts-config", "/home/model-server/config.properties", "--model-store", "/home/model-server/"]