Skip to content

Commit 628ecfe

Browse files
committed
Revert changes in building onnx on gpu (build on buster, not on ubuntu16)
1 parent 47d222b commit 628ecfe

File tree

2 files changed

+5
-20
lines changed

2 files changed

+5
-20
lines changed

opt/build/onnxruntime/Dockerfile.x64

Lines changed: 1 addition & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -10,18 +10,10 @@ ARG ONNXRUNTIME_VER=1.7.1
1010
ARG ARCH=x64
1111

1212
RUN apt-get -qq update
13-
RUN apt-get install -y software-properties-common
14-
RUN add-apt-repository ppa:deadsnakes/ppa
15-
RUN apt-get -qq update
16-
17-
RUN apt-get -qq install -y python3.6 curl wget tar git \
13+
RUN apt-get -qq install -y curl wget tar git \
1814
build-essential cmake \
1915
libcurl4-openssl-dev libssl-dev libatlas-base-dev zlib1g-dev \
2016
python3 python3-pip python3-dev python3-numpy
21-
22-
RUN rm /usr/bin/python3
23-
RUN ln -s /usr/bin/python3.6 /usr/bin/python3
24-
2517
RUN pip3 install --upgrade pip setuptools wheel
2618

2719
ENV LANG=en_US.UTF-8

opt/build/onnxruntime/Dockerfile.x64-gpu

Lines changed: 4 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -1,26 +1,19 @@
1-
ARG OS=ubuntu16.04
1+
ARG OS=ubuntu18.04
22
ARG CUDA_VER=11.0-cudnn8
33

4-
FROM nvidia/cuda:11.0-cudnn8-devel-ubuntu16.04
4+
FROM nvidia/cuda:11.0-cudnn8-devel-ubuntu18.04
55

66
ARG ONNXRUNTIME_REPO=https://github.com/RedisAI/onnxruntime
77
ARG ONNXRUNTIME_VER=1.7.1
88
ARG ARCH=x64-gpu
99

1010
RUN apt-get -qq update
11-
RUN apt-get install -y software-properties-common
12-
RUN add-apt-repository ppa:deadsnakes/ppa
13-
RUN apt-get -qq update
14-
15-
RUN apt-get -qq install -y python3.6 curl \
11+
RUN apt-get -qq install -y curl \
1612
wget tar git build-essential \
1713
libcurl4-openssl-dev libssl-dev \
1814
libatlas-base-dev zlib1g-dev python3 python3-pip \
1915
python3-dev python3-numpy rsync
2016

21-
RUN rm /usr/bin/python3
22-
RUN ln -s /usr/bin/python3.6 /usr/bin/python3
23-
2417
RUN pip3 install --upgrade pip setuptools wheel
2518

2619
RUN wget -q https://github.com/Kitware/CMake/releases/download/v3.19.5/cmake-3.19.5-Linux-x86_64.tar.gz -O /tmp/cmake.tgz
@@ -42,7 +35,7 @@ ARG BUILDARGS="--config ${BUILDTYPE} --parallel"
4235

4336
RUN git clone --single-branch --branch rel-${ONNXRUNTIME_VER} ${ONNXRUNTIME_REPO} onnxruntime
4437
WORKDIR /build/onnxruntime
45-
RUN git fetch --recurse-submodules
38+
RUN git fetch --recurse-submodules -j4
4639
RUN ./build.sh ${BUILDARGS} --update --build --use_cuda --cudnn_home /usr/local/cuda --cuda_home /usr/local/cuda --build_shared_lib --parallel
4740

4841
ADD ./pack.sh /build/

0 commit comments

Comments
 (0)