Hey folks, while there are already a few solutions that exist to show Nvidia GPU stats via MQTT (and therefore in HA), I wanted to be able to see what processes were running on the GPU so that I could track which ones were consuming what.
I’d already created a Docker container which provided a web interface to show stats and runtimes, so I extended it to also report via MQTT.
You can check it out here: GitHub - loryanstrant/GPU-model-monitor-MQTT: A Docker container that monitors your GPU and gives details around model usage (e.g. nvidia-smi), via a web interface as well as MQTT / Home Assistant.
The screenshots are a bit out of date, as I added some functionality to show percentage of GPU used as well as average & max values.