diff --git a/Dockerfile.train.tmpl b/Dockerfile.train.tmpl
index 1cc3dffc..7feab79e 100644
--- a/Dockerfile.train.tmpl
+++ b/Dockerfile.train.tmpl
@@ -46,4 +46,8 @@ RUN pip3 install --upgrade pip==20.0.2 wheel==0.34.2 setuptools==46.1.3
 #    we don't want to break that
 RUN DS_NODECODER=y DS_NOTENSORFLOW=y pip3 install --upgrade -e .
 
+# Tool to convert output graph for inference
+RUN python3 util/taskcluster.py --source tensorflow --branch r1.15 \
+        --artifact convert_graphdef_memmapped_format  --target .
+
 RUN ./bin/run-ldc93s1.sh