ARG BASE_IMAGE
FROM ${BASE_IMAGE}
USER root
# These are copies of the existing locked ONNX environment, not downloads.
COPY python/ /opt/contextra-model-python/
COPY engine_contract.cpp /opt/contextra-model-tools/engine_contract.cpp
RUN g++ -std=c++17 -O2 /opt/contextra-model-tools/engine_contract.cpp \
    -I/usr/local/cuda/include -lnvinfer -lnvinfer_plugin \
    -o /usr/local/bin/contextra-engine-contract
ENV PYTHONPATH=/opt/contextra-model-python:/opt/contextra
ENV CONTEXTRA_ENGINE_PROBE=/usr/local/bin/contextra-engine-contract
RUN python3 -B -c 'import onnx,numpy,tritonclient.grpc; print(onnx.__version__)'