fix: Update the dockerfile for pytorch converter

733ea0b2 · Zhang Zhimin · 6c9f380e · 733ea0b2
隐藏空白更改
内联并排

Showing with 3 addition and 0 deletion

docker/mace-dev/Dockerfile docker/mace-dev/Dockerfile +3 -0

未找到文件。
--- 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 && \