Fix kubeflowkatib/mxnet-mnist image (#1866)

kubeflow · May 18, 2022 · d385d14 · d385d14
1 parent ea23e71
commit d385d14
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/examples/v1beta1/trial-images/mxnet-mnist/Dockerfile b/examples/v1beta1/trial-images/mxnet-mnist/Dockerfile
@@ -1,8 +1,9 @@
-FROM mxnet/python:latest_cpu_native_py3
+FROM python:3.9
 
 ADD examples/v1beta1/trial-images/mxnet-mnist /opt/mxnet-mnist
 WORKDIR /opt/mxnet-mnist
 
+RUN pip install mxnet==1.9.0
 RUN chgrp -R 0 /opt/mxnet-mnist \
   && chmod -R g+rwX /opt/mxnet-mnist