Add nvidia-smi function

Signed-off-by: Han Verstraete (OpenFaaS Ltd) <[email protected]>

Author: welteki

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

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