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