diff --git a/Dockerfile.seldon b/Dockerfile.seldon new file mode 100644 index 000000000..11f31d273 --- /dev/null +++ b/Dockerfile.seldon @@ -0,0 +1,35 @@ +FROM axsauze/vulkan-sum:0.1 as vulkan-base + +# No need to do any work + +FROM seldonio/seldon-core-s2i-python3:1.1.1-rc + +ENV NVIDIA_DRIVER_CAPABILITIES compute,graphics,utility + +RUN apt-get update && apt-get install -y --no-install-recommends \ + libx11-xcb-dev \ + libxkbcommon-dev \ + libwayland-dev \ + libxrandr-dev \ + libegl1-mesa-dev && \ + rm -rf /var/lib/apt/lists/* + +COPY --from=vulkan-base /usr/local/bin /usr/local/bin +COPY --from=vulkan-base /usr/local/lib /usr/local/lib +COPY --from=vulkan-base /usr/local/include/vulkan /usr/local/include/vulkan +COPY --from=vulkan-base /usr/local/share/vulkan /usr/local/share/vulkan + +COPY /etc/vulkan/icd.d/nvidia_icd.json /etc/vulkan/icd.d/nvidia_icd.json + + +# Seldon Env Vars +ENV MODEL_NAME Model +ENV API_TYPE REST +ENV SERVICE_TYPE MODEL +ENV PERSISTENCE 0 + +COPY bin/ bin/ +COPY shaders/ shaders/ +COPY seldon/Model.py Model.py + + diff --git a/seldon/Model.py b/seldon/Model.py new file mode 100644 index 000000000..f9c0417d7 --- /dev/null +++ b/seldon/Model.py @@ -0,0 +1,8 @@ +import subprocess + +class Model: + def __init__(self): + pass + + def predict(self, X, names=[], meta=[]): + return subprocess.check_output(["bin/main"])