Fix models images (#450)

mlrun · Sep 26, 2020 · 62f4355 · 62f4355
1 parent 0a71cb0
commit 62f4355
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/dockerfiles/models/Dockerfile b/dockerfiles/models/Dockerfile
@@ -57,7 +57,7 @@ RUN conda clean -ayq
 RUN python -m pip install torch==1.4.0+cpu torchvision==0.5.0+cpu \
     -f https://download.pytorch.org/whl/torch_stable.html
 
-RUN python -m pip install -U intel-tensorflow
+RUN python -m pip install -U intel-tensorflow mxnet
 
 RUN HOROVOD_WITH_TENSORFLOW=1 HOROVOD_WITH_PYTORCH=1 \
     python -m pip install horovod

diff --git a/dockerfiles/models/py36/Dockerfile b/dockerfiles/models/py36/Dockerfile
@@ -58,7 +58,7 @@ RUN conda install -y -c plotly plotly-orca python=3.6
 RUN python -m pip install torch==1.4.0+cpu torchvision==0.5.0+cpu \
     -f https://download.pytorch.org/whl/torch_stable.html
 
-RUN python -m pip install -U 'intel-tensorflow<2' keras
+RUN python -m pip install -U 'intel-tensorflow<2' keras mxnet
 
 RUN HOROVOD_WITH_TENSORFLOW=1 HOROVOD_WITH_PYTORCH=1 \
     python -m pip install horovod