diff --git a/Dockerfile b/Dockerfile
index 416404f243ff1bdd4fd36cad9b761a17bfd3311a..4020cf467cc0a79f8b439cc783a01727160c4803 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -107,9 +107,11 @@ RUN apt-get install -y --no-install-recommends \
     android-tools-adb
 
 # Install tools
-RUN pip install tensorflow==1.4.0 && \
+RUN pip install setuptools && \
+    pip install tensorflow==1.4.0 && \
     pip install scipy && \
     pip install jinja2
 
 # Download tensorflow tools
-RUN wget http://cnbj1-inner-fds.api.xiaomi.net/mace/tool/transform_graph
+RUN wget http://cnbj1-inner-fds.api.xiaomi.net/mace/tool/transform_graph && \
+    chmod +x transform_graph