Added seldon
This commit is contained in:
parent
dc3d94dd2a
commit
4a54a37590
2 changed files with 43 additions and 0 deletions
35
Dockerfile.seldon
Normal file
35
Dockerfile.seldon
Normal 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
8
seldon/Model.py
Normal 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"])
|
||||||
Loading…
Add table
Add a link
Reference in a new issue