diff --git a/nvidia-smi/Dockerfile b/nvidia-smi/Dockerfile new file mode 100644 index 0000000..a3fa20a --- /dev/null +++ b/nvidia-smi/Dockerfile @@ -0,0 +1,13 @@ +FROM ghcr.io/openfaas/classic-watchdog:0.3.2 AS watchdog + +FROM ubuntu:24.04 + +COPY --from=watchdog /fwatchdog /usr/bin/fwatchdog +RUN chmod +x /usr/bin/fwatchdog + +USER 1000 + +ENV fprocess="nvidia-smi" + +HEALTHCHECK --interval=3s CMD [ -e /tmp/.lock ] || exit 1 +CMD ["fwatchdog"] diff --git a/stack.yml b/stack.yml index 9e78c55..0c135f9 100644 --- a/stack.yml +++ b/stack.yml @@ -100,11 +100,14 @@ functions: lang: golang-middleware handler: ./markdown image: ${SERVER:-ghcr.io}/${OWNER:-openfaas}/markdown-fn:${TAG:-latest} + + nvidia-smi: + lang: dockerfile + handler: ./nvidia-smi + image: ${SERVER:-ghcr.io}/${OWNER:-openfaas}/nvidia-smi:${TAG:-latest} configuration: templates: - name: golang-middleware source: https://github.com/openfaas/golang-http-template - -