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