diff --git a/docker/1.2.0/py2/Dockerfile.cpu b/docker/1.2.0/py2/Dockerfile.cpu index 3429783b..e14be3b2 100644 --- a/docker/1.2.0/py2/Dockerfile.cpu +++ b/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"] \ No newline at end of file +CMD ["mxnet-model-server", "--start", "--mms-config", "/home/model-server/config.properties"] diff --git a/docker/1.2.0/py2/Dockerfile.gpu b/docker/1.2.0/py2/Dockerfile.gpu index b7c8a92a..25cfe3ba 100644 --- a/docker/1.2.0/py2/Dockerfile.gpu +++ b/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"] \ No newline at end of file +CMD ["mxnet-model-server", "--start", "--mms-config", "/home/model-server/config.properties"] diff --git a/docker/1.2.0/py3/Dockerfile.cpu b/docker/1.2.0/py3/Dockerfile.cpu index 86e2fb32..2a498e01 100644 --- a/docker/1.2.0/py3/Dockerfile.cpu +++ b/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"] \ No newline at end of file +CMD ["mxnet-model-server", "--start", "--mms-config", "/home/model-server/config.properties"] diff --git a/docker/1.2.0/py3/Dockerfile.gpu b/docker/1.2.0/py3/Dockerfile.gpu index 3315e28b..559d8567 100644 --- a/docker/1.2.0/py3/Dockerfile.gpu +++ b/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"] \ No newline at end of file +CMD ["mxnet-model-server", "--start", "--mms-config", "/home/model-server/config.properties"]