Skip to content

Commit 2fbd915

Browse files
committed
Build onnx on ubuntu 16 instead of buster
1 parent d6d98ef commit 2fbd915

File tree

2 files changed

+8
-4
lines changed

2 files changed

+8
-4
lines changed

opt/build/onnxruntime/Dockerfile.x64

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
11
# BUILD redisfab/onnxruntime-x64-${OS}:$(ONNXRUNTIME_VER)
22

3-
ARG OS=debian:buster
3+
ARG OS=ubuntu:16.04
44

55
#----------------------------------------------------------------------------------------------
66
FROM ${OS}

opt/build/onnxruntime/Dockerfile.x64-gpu

Lines changed: 7 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,13 +1,17 @@
1-
ARG OS=ubuntu18.04
1+
ARG OS=ubuntu16.04
22
ARG CUDA_VER=11.0-cudnn8
33

4-
FROM nvidia/cuda:11.0-cudnn8-devel-ubuntu18.04
4+
FROM nvidia/cuda:11.0-cudnn8-devel-ubuntu16.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+
14+
RUN apt-get install -y python3.6
1115
RUN apt-get -qq install -y curl \
1216
wget tar git build-essential \
1317
libcurl4-openssl-dev libssl-dev \
@@ -35,7 +39,7 @@ ARG BUILDARGS="--config ${BUILDTYPE} --parallel"
3539

3640
RUN git clone --single-branch --branch rel-${ONNXRUNTIME_VER} ${ONNXRUNTIME_REPO} onnxruntime
3741
WORKDIR /build/onnxruntime
38-
RUN git fetch --recurse-submodules -j4
42+
RUN git fetch --recurse-submodules
3943
RUN ./build.sh ${BUILDARGS} --update --build --use_cuda --cudnn_home /usr/local/cuda --cuda_home /usr/local/cuda --build_shared_lib --parallel
4044

4145
ADD ./pack.sh /build/

0 commit comments

Comments
 (0)