diff --git a/examples/mnist/Dockerfile b/examples/mnist/Dockerfile
index b4e02ced1..7e228b7f9 100644
--- a/examples/mnist/Dockerfile
+++ b/examples/mnist/Dockerfile
@@ -1,4 +1,4 @@
-FROM pytorch/pytorch:1.0-cuda10.0-cudnn7-runtime
+FROM kubeflow/pytorch:1.0-cuda10.0-cudnn7-runtime
 
 RUN pip install tensorboardX==1.6.0
 WORKDIR /var