File tree Expand file tree Collapse file tree 2 files changed +8
-4
lines changed
Expand file tree Collapse file tree 2 files changed +8
-4
lines changed Original file line number Diff line number Diff line change 11# BUILD redisfab/onnxruntime-x64-${OS}:$(ONNXRUNTIME_VER)
22
3- ARG OS=debian:buster
3+ ARG OS=ubuntu:16.04
44
55#----------------------------------------------------------------------------------------------
66FROM ${OS}
Original file line number Diff line number Diff line change 1- ARG OS=ubuntu18 .04
1+ ARG OS=ubuntu16 .04
22ARG 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
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+
14+ RUN apt-get install -y python3.6
1115RUN 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
3640RUN git clone --single-branch --branch rel-${ONNXRUNTIME_VER} ${ONNXRUNTIME_REPO} onnxruntime
3741WORKDIR /build/onnxruntime
38- RUN git fetch --recurse-submodules -j4
42+ RUN git fetch --recurse-submodules
3943RUN ./build.sh ${BUILDARGS} --update --build --use_cuda --cudnn_home /usr/local/cuda --cuda_home /usr/local/cuda --build_shared_lib --parallel
4044
4145ADD ./pack.sh /build/
You can’t perform that action at this time.
0 commit comments