@@ -2,16 +2,6 @@ FROM nvidia/cuda:10.1-cudnn7-devel
2
2
3
3
ARG ONNXRUNTIME_VERSION="0.5.0"
4
4
5
- RUN apt-get update -qq && apt-get install -y -q \
6
- python3 \
7
- python3-dev \
8
- python3-pip \
9
- && apt-get clean -qq && rm -rf /var/lib/apt/lists/* && \
10
- pip3 install --upgrade \
11
- pip \
12
- setuptools \
13
- && rm -rf /root/.cache/pip*
14
-
15
5
RUN apt-get update -qq && apt-get install -y -q \
16
6
build-essential \
17
7
curl \
@@ -23,16 +13,20 @@ RUN apt-get update -qq && apt-get install -y -q \
23
13
software-properties-common \
24
14
unzip \
25
15
zlib1g-dev \
26
- && apt-get clean -qq && rm -rf /var/lib/apt/lists/*
27
-
16
+ python3.6-dev \
17
+ python3.6-distutils \
18
+ && apt-get clean -qq && rm -rf /var/lib/apt/lists/* && \
19
+ curl https://bootstrap.pypa.io/get-pip.py -o get-pip.py && \
20
+ python3.6 get-pip.py && \
21
+ rm -rf /root/.cache/pip*
28
22
29
23
ENV PYTHONPATH "${PYTHONPATH}:/src:/mnt/project"
30
24
31
25
COPY pkg/workloads/cortex/lib/requirements.txt /src/cortex/lib/requirements.txt
32
26
COPY pkg/workloads/cortex/onnx_serve/requirements.txt /src/cortex/onnx_serve/requirements.txt
33
- RUN pip3 install -r /src/cortex/lib/requirements.txt && \
34
- pip3 install -r /src/cortex/onnx_serve/requirements.txt && \
35
- pip3 install onnxruntime-gpu==${ONNXRUNTIME_VERSION} && \
27
+ RUN pip install -r /src/cortex/lib/requirements.txt && \
28
+ pip install -r /src/cortex/onnx_serve/requirements.txt && \
29
+ pip install onnxruntime-gpu==${ONNXRUNTIME_VERSION} && \
36
30
rm -rf /root/.cache/pip*
37
31
38
32
COPY pkg/workloads/cortex/consts.py /src/cortex
0 commit comments