From 733ea0b2adc32a247c856cca6940403720f93bb4 Mon Sep 17 00:00:00 2001
From: Zhang Zhimin <zhangzhimin1@xiaomi.com>
Date: Mon, 2 Nov 2020 11:48:51 +0800
Subject: [PATCH] fix: Update the dockerfile for pytorch converter

---
 docker/mace-dev/Dockerfile | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/docker/mace-dev/Dockerfile b/docker/mace-dev/Dockerfile
index 5ae643a9..c2686b9c 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 && \
-- 
GitLab