Added seldon

This commit is contained in:
Alejandro Saucedo 2020-08-05 08:12:51 +01:00
parent dc3d94dd2a
commit 4a54a37590
2 changed files with 43 additions and 0 deletions

35
Dockerfile.seldon Normal file
View file

@ -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

8
seldon/Model.py Normal file
View file

@ -0,0 +1,8 @@
import subprocess
class Model:
def __init__(self):
pass
def predict(self, X, names=[], meta=[]):
return subprocess.check_output(["bin/main"])