From 4a54a37590142951975481cd3ffd4e2f2c35cb8a Mon Sep 17 00:00:00 2001
From: Alejandro Saucedo <axsauze@gmail.com>
Date: Wed, 5 Aug 2020 08:12:51 +0100
Subject: [PATCH] Added seldon

---
 Dockerfile.seldon | 35 +++++++++++++++++++++++++++++++++++
 seldon/Model.py   |  8 ++++++++
 2 files changed, 43 insertions(+)
 create mode 100644 Dockerfile.seldon
 create mode 100644 seldon/Model.py

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"])