Upgrade MMS to v1.0.7 and reduce CUDA packages (#6)

Author: nikhil-sk

docker/1.2.0/py2/Dockerfile.cpu

@@ -43,7 +43,7 @@ ENV PATH /opt/conda/bin:$PATH
 
 ARG PYTORCH_VERSION=1.2.0
 ARG TORCHVISION_VERSION=0.4.0
-ARG MMS_VERSION=1.0.5
+ARG MMS_VERSION=1.0.7
 RUN conda install -c conda-forge awscli==1.16.210 opencv==4.0.1 && \
     conda install -y scikit-learn==0.20.3 \
                      pandas==0.24.2 \
@@ -53,7 +53,7 @@ RUN conda install -c conda-forge awscli==1.16.210 opencv==4.0.1 && \
     conda install pytorch==$PYTORCH_VERSION torchvision==$TORCHVISION_VERSION cpuonly -c pytorch && \
     conda clean -ya && \
     pip install --upgrade pip --trusted-host pypi.org --trusted-host files.pythonhosted.org && \
-    pip install --pre -U mxnet-model-server
+    pip install mxnet-model-server==$MMS_VERSION
 
 RUN useradd -m model-server \
     && mkdir -p /home/model-server/tmp \
@@ -75,4 +75,4 @@ ENV SAGEMAKER_SERVING_MODULE sagemaker_pytorch_serving_container.serving:main
 EXPOSE 8080 8081
 ENV TEMP=/home/model-server/tmp
 ENTRYPOINT ["python", "/usr/local/bin/dockerd-entrypoint.py"]
-CMD ["mxnet-model-server", "--start", "--mms-config", "/home/model-server/config.properties"]
+CMD ["mxnet-model-server", "--start", "--mms-config", "/home/model-server/config.properties"]

docker/1.2.0/py2/Dockerfile.gpu

@@ -6,12 +6,6 @@ RUN apt-get update && apt-get install -y  --allow-downgrades --allow-change-held
         build-essential \
         ca-certificates \
         cmake \
-        cuda-command-line-tools-10-0 \
-        cuda-cublas-10-0 \
-        cuda-cufft-10-0 \
-        cuda-curand-10-0 \
-        cuda-cusolver-10-0 \
-        cuda-cusparse-10-0 \
         libgomp1 \
         libibverbs-dev \
         curl \
@@ -59,7 +53,7 @@ ENV PATH /opt/conda/bin:$PATH
 
 ARG PYTORCH_VERSION=1.2.0
 ARG TORCHVISION_VERSION=0.4.0
-ARG MMS_VERSION=1.0.5
+ARG MMS_VERSION=1.0.7
 RUN conda install -c pytorch magma-cuda100 && \
     conda install -c conda-forge awscli==1.16.210 opencv==4.0.1 && \
     conda install -y scikit-learn==0.20.3 \
@@ -71,7 +65,7 @@ RUN conda install -c pytorch magma-cuda100 && \
     conda clean -ya && \
     /opt/conda/bin/conda config --set ssl_verify False && \
     pip install --upgrade pip --trusted-host pypi.org --trusted-host files.pythonhosted.org && \
-    pip install --pre -U mxnet-model-server
+    pip install mxnet-model-server==$MMS_VERSION
 
 RUN useradd -m model-server \
     && mkdir -p /home/model-server/tmp \
@@ -102,4 +96,4 @@ ENV SAGEMAKER_SERVING_MODULE sagemaker_pytorch_serving_container.serving:main
 EXPOSE 8080 8081
 ENV TEMP=/home/model-server/tmp
 ENTRYPOINT ["python", "/usr/local/bin/dockerd-entrypoint.py"]
-CMD ["mxnet-model-server", "--start", "--mms-config", "/home/model-server/config.properties"]
+CMD ["mxnet-model-server", "--start", "--mms-config", "/home/model-server/config.properties"]

docker/1.2.0/py3/Dockerfile.cpu

@@ -43,7 +43,7 @@ ENV PATH /opt/conda/bin:$PATH
 
 ARG PYTORCH_VERSION=1.2.0
 ARG TORCHVISION_VERSION=0.4.0
-ARG MMS_VERSION=1.0.5
+ARG MMS_VERSION=1.0.7
 RUN conda install -c conda-forge awscli==1.16.210 opencv==4.0.1 && \
     conda install -y scikit-learn==0.21.2 \
                      pandas==0.25.0 \
@@ -54,7 +54,7 @@ RUN conda install -c conda-forge awscli==1.16.210 opencv==4.0.1 && \
     conda clean -ya && \
     pip install --upgrade pip --trusted-host pypi.org --trusted-host files.pythonhosted.org && \
     ln -s /opt/conda/bin/pip /usr/local/bin/pip3 && \
-    pip install --pre -U mxnet-model-server
+    pip install mxnet-model-server==$MMS_VERSION
 
 RUN useradd -m model-server \
     && mkdir -p /home/model-server/tmp \
@@ -76,4 +76,4 @@ ENV SAGEMAKER_SERVING_MODULE sagemaker_pytorch_serving_container.serving:main
 EXPOSE 8080 8081
 ENV TEMP=/home/model-server/tmp
 ENTRYPOINT ["python", "/usr/local/bin/dockerd-entrypoint.py"]
-CMD ["mxnet-model-server", "--start", "--mms-config", "/home/model-server/config.properties"]
+CMD ["mxnet-model-server", "--start", "--mms-config", "/home/model-server/config.properties"]

docker/1.2.0/py3/Dockerfile.gpu

@@ -6,12 +6,6 @@ RUN apt-get update && apt-get install -y  --allow-downgrades --allow-change-held
         build-essential \
         ca-certificates \
         cmake \
-        cuda-command-line-tools-10-0 \
-        cuda-cublas-10-0 \
-        cuda-cufft-10-0 \
-        cuda-curand-10-0 \
-        cuda-cusolver-10-0 \
-        cuda-cusparse-10-0 \
         libgomp1 \
         libibverbs-dev \
         curl \
@@ -59,7 +53,7 @@ ENV PATH /opt/conda/bin:$PATH
 
 ARG PYTORCH_VERSION=1.2.0
 ARG TORCHVISION_VERSION=0.4.0
-ARG MMS_VERSION=1.0.5
+ARG MMS_VERSION=1.0.7
 RUN conda install -c pytorch magma-cuda100 && \
     conda install -c conda-forge awscli==1.16.210 opencv==4.0.1 && \
     conda install -y scikit-learn==0.21.2 \
@@ -72,7 +66,7 @@ RUN conda install -c pytorch magma-cuda100 && \
     /opt/conda/bin/conda config --set ssl_verify False && \
     pip install --upgrade pip --trusted-host pypi.org --trusted-host files.pythonhosted.org && \
     ln -s /opt/conda/bin/pip /usr/local/bin/pip3 && \
-    pip install --pre -U mxnet-model-server
+    pip install mxnet-model-server==$MMS_VERSION
 
 RUN useradd -m model-server \
     && mkdir -p /home/model-server/tmp \
@@ -94,4 +88,4 @@ ENV SAGEMAKER_SERVING_MODULE sagemaker_pytorch_serving_container.serving:main
 EXPOSE 8080 8081
 ENV TEMP=/home/model-server/tmp
 ENTRYPOINT ["python", "/usr/local/bin/dockerd-entrypoint.py"]
-CMD ["mxnet-model-server", "--start", "--mms-config", "/home/model-server/config.properties"]
+CMD ["mxnet-model-server", "--start", "--mms-config", "/home/model-server/config.properties"]