diff --git a/docker/mace-dev/Dockerfile b/docker/mace-dev/Dockerfile
index 5ae643a96e1ca49732982182d52d38ee1936a463..c2686b9ceb7c11e93d270c5615851296e306b232 100644
--- a/docker/mace-dev/Dockerfile
+++ b/docker/mace-dev/Dockerfile
@@ -17,6 +17,9 @@ RUN pip install -i http://mirrors.aliyun.com/pypi/simple/ --trusted-host mirrors
     sphinx_rtd_theme \
     recommonmark
 
+RUN pip install -i http://mirrors.aliyun.com/pypi/simple/ --trusted-host mirrors.aliyun.com \
+    torch==1.4.0 onnx onnx_tf
+
 # Customize vim
 RUN git clone --recursive https://github.com/llhe/vimrc.git && \
     cd vimrc && \