You are viewing a plain text version of this content. The canonical link for it is here.
Posted to commits@singa.apache.org by wa...@apache.org on 2020/03/31 14:05:12 UTC
[singa] branch dev updated: Add onnx 1.6.0 in docker
This is an automated email from the ASF dual-hosted git repository.
wangwei pushed a commit to branch dev
in repository https://gitbox.apache.org/repos/asf/singa.git
The following commit(s) were added to refs/heads/dev by this push:
new df46860 Add onnx 1.6.0 in docker
new 4ef1b79 Merge pull request #640 from chrishkchris/remove_conda-forge
df46860 is described below
commit df468600580c2c968e594293f7dbbd7deeab73c4
Author: Chris Yeung <ch...@yahoo.com.hk>
AuthorDate: Tue Mar 31 15:04:55 2020 +0800
Add onnx 1.6.0 in docker
---
tool/docker/devel/ubuntu/cuda10/Dockerfile | 6 ++++--
tool/docker/devel/ubuntu/cuda9/Dockerfile | 5 +++--
2 files changed, 7 insertions(+), 4 deletions(-)
diff --git a/tool/docker/devel/ubuntu/cuda10/Dockerfile b/tool/docker/devel/ubuntu/cuda10/Dockerfile
index 84e571c..02d546a 100644
--- a/tool/docker/devel/ubuntu/cuda10/Dockerfile
+++ b/tool/docker/devel/ubuntu/cuda10/Dockerfile
@@ -19,6 +19,7 @@
FROM nvidia/cuda:10.0-devel-ubuntu18.04
ENV CUDNN_VERSION 7.4.2.24
+
RUN apt-get update && apt-get install -y --no-install-recommends \
libcudnn7=$CUDNN_VERSION-1+cuda10.0 \
libcudnn7-dev=$CUDNN_VERSION-1+cuda10.0 && \
@@ -48,13 +49,14 @@ RUN apt-get update \
&& apt-get autoclean \
&& rm -rf /var/lib/apt/lists/* \
&& pip3 install -U --no-cache \
- && pip3 install tqdm \
+ tqdm \
wheel \
numpy \
setuptools \
protobuf \
Deprecated \
- future
+ future \
+ onnx==1.6.0
# install cmake to correctly find Cuda 10
RUN wget https://github.com/Kitware/CMake/releases/download/v3.12.2/cmake-3.12.2.tar.gz -P /tmp/ \
diff --git a/tool/docker/devel/ubuntu/cuda9/Dockerfile b/tool/docker/devel/ubuntu/cuda9/Dockerfile
index b7678c7..5d59dbf 100644
--- a/tool/docker/devel/ubuntu/cuda9/Dockerfile
+++ b/tool/docker/devel/ubuntu/cuda9/Dockerfile
@@ -49,13 +49,14 @@ RUN apt-get update \
&& apt-get autoclean \
&& rm -rf /var/lib/apt/lists/* \
&& pip3 install -U --no-cache \
- && pip3 install tqdm \
+ tqdm \
wheel \
numpy \
setuptools \
protobuf \
Deprecated \
- future
+ future \
+ onnx==1.6.0
# install swig > 3.0.10
RUN wget http://prdownloads.sourceforge.net/swig/swig-3.0.10.tar.gz -P /tmp/ \