From 05798c6a62ab73c1366a502bafb1bd08065cab2b Mon Sep 17 00:00:00 2001
From: craig-warren <craig_warren@me.com>
Date: Sun, 10 Dec 2023 22:30:52 +0000
Subject: [PATCH] Re-add nvidia smi message

---
 gprMax/model_build_run.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/gprMax/model_build_run.py b/gprMax/model_build_run.py
index 93e812bd..397682c5 100644
--- a/gprMax/model_build_run.py
+++ b/gprMax/model_build_run.py
@@ -352,6 +352,7 @@ class ModelBuildRun:
         logger.info(
             f"Time taken: " + f"{humanize.precisedelta(datetime.timedelta(seconds=solver.solvetime), format='%0.4f')}"
         )
+        logger.basic(subprocess.check_output(["nvidia-smi"], shell=False, stderr=subprocess.STDOUT))
 
 
 class GridBuilder: