From 42bb9ea2e75f8a615438038015495d1b858a255f Mon Sep 17 00:00:00 2001 From: Ofer Hasson Date: Sun, 26 Apr 2020 16:18:15 +0300 Subject: [PATCH] Update Docker file to MXNet 1.6 Since the main pip package suggests the latest MXNet version, the containers should be on par --- docker/Dockerfile.cpu | 2 +- docker/Dockerfile.gpu | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/docker/Dockerfile.cpu b/docker/Dockerfile.cpu index 316cb5479..77b26f117 100644 --- a/docker/Dockerfile.cpu +++ b/docker/Dockerfile.cpu @@ -21,7 +21,7 @@ RUN update-alternatives --install /usr/bin/python python /usr/bin/python3 1 RUN update-alternatives --install /usr/local/bin/pip pip /usr/local/bin/pip3 1 RUN pip install --no-cache-dir multi-model-server \ - && pip install --no-cache-dir mxnet-mkl==1.4.0 + && pip install --no-cache-dir mxnet-mkl==1.6.0 RUN useradd -m model-server \ && mkdir -p /home/model-server/tmp diff --git a/docker/Dockerfile.gpu b/docker/Dockerfile.gpu index 112d3248d..5852f8d7b 100644 --- a/docker/Dockerfile.gpu +++ b/docker/Dockerfile.gpu @@ -21,7 +21,7 @@ RUN update-alternatives --install /usr/bin/python python /usr/bin/python3 1 RUN update-alternatives --install /usr/local/bin/pip pip /usr/local/bin/pip3 1 RUN pip install --no-cache-dir multi-model-server \ - && pip install --no-cache-dir mxnet-cu92mkl==1.4.0 + && pip install --no-cache-dir mxnet-cu92mkl==1.6.0 RUN useradd -m model-server \ && mkdir -p /home/model-server/tmp