2021-04-20 13:50:41 +02:00
|
|
|
ARG FROM_IMAGE_NAME=nvcr.io/nvidia/tensorflow:20.12-tf1-py3
|
|
|
|
ARG TRITON_CLIENT_IMAGE_NAME=nvcr.io/nvidia/tritonserver:20.12-py3-sdk
|
|
|
|
FROM ${TRITON_CLIENT_IMAGE_NAME} as triton-client
|
2020-06-12 12:38:25 +02:00
|
|
|
FROM ${FROM_IMAGE_NAME}
|
|
|
|
|
2021-04-20 13:50:41 +02:00
|
|
|
# Install perf_client required library
|
|
|
|
RUN apt-get update && \
|
|
|
|
apt-get install -y libb64-dev libb64-0d && \
|
|
|
|
apt-get clean && \
|
|
|
|
rm -rf /var/lib/apt/lists/*
|
|
|
|
|
|
|
|
# Install Triton Client PythonAPI and copy Perf Client
|
|
|
|
COPY --from=triton-client /workspace/install/ /workspace/install/
|
|
|
|
ENV LD_LIBRARY_PATH /workspace/install/lib:${LD_LIBRARY_PATH}
|
|
|
|
RUN find /workspace/install/python/ -iname triton*manylinux*.whl -exec pip install {}[all] \;
|
|
|
|
|
|
|
|
# Setup environmnent variables to access Triton Client lib and bin
|
|
|
|
ENV PATH /workspace/install/bin:${PATH}
|
|
|
|
|
|
|
|
ENV PYTHONPATH /workspace/rn50v15_tf
|
|
|
|
WORKDIR /workspace/rn50v15_tf
|
|
|
|
|
|
|
|
RUN pip uninstall -y typing
|
|
|
|
|
2020-06-12 12:38:25 +02:00
|
|
|
ADD requirements.txt .
|
2021-04-20 13:50:41 +02:00
|
|
|
ADD triton/requirements.txt triton/requirements.txt
|
2020-06-12 12:38:25 +02:00
|
|
|
RUN pip install -r requirements.txt
|
2021-04-20 13:50:41 +02:00
|
|
|
RUN pip install -r triton/requirements.txt
|
2020-06-12 12:38:25 +02:00
|
|
|
|
2021-04-20 13:50:41 +02:00
|
|
|
ADD . .
|