File tree Expand file tree Collapse file tree 2 files changed +5
-20
lines changed
Expand file tree Collapse file tree 2 files changed +5
-20
lines changed Original file line number Diff line number Diff line change @@ -10,18 +10,10 @@ ARG ONNXRUNTIME_VER=1.7.1
1010ARG ARCH=x64
1111
1212RUN 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-
2517RUN pip3 install --upgrade pip setuptools wheel
2618
2719ENV LANG=en_US.UTF-8
Original file line number Diff line number Diff line change 1- ARG OS=ubuntu16 .04
1+ ARG OS=ubuntu18 .04
22ARG 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
66ARG ONNXRUNTIME_REPO=https://github.com/RedisAI/onnxruntime
77ARG ONNXRUNTIME_VER=1.7.1
88ARG ARCH=x64-gpu
99
1010RUN 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-
2417RUN pip3 install --upgrade pip setuptools wheel
2518
2619RUN 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
4336RUN git clone --single-branch --branch rel-${ONNXRUNTIME_VER} ${ONNXRUNTIME_REPO} onnxruntime
4437WORKDIR /build/onnxruntime
45- RUN git fetch --recurse-submodules
38+ RUN git fetch --recurse-submodules -j4
4639RUN ./build.sh ${BUILDARGS} --update --build --use_cuda --cudnn_home /usr/local/cuda --cuda_home /usr/local/cuda --build_shared_lib --parallel
4740
4841ADD ./pack.sh /build/
You can’t perform that action at this time.
0 commit comments