From 38661d586874dd28403139e46d2711707f63f44b Mon Sep 17 00:00:00 2001
From: craig-warren <craig_warren@me.com>
Date: Sun, 10 Dec 2023 21:10:30 +0000
Subject: [PATCH] Some debugging for CUDA mem leak

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

diff --git a/gprMax/model_build_run.py b/gprMax/model_build_run.py
index 622d2886..1bb020a5 100644
--- a/gprMax/model_build_run.py
+++ b/gprMax/model_build_run.py
@@ -19,6 +19,7 @@
 import datetime
 import itertools
 import logging
+import subprocess
 import sys
 
 import humanize
@@ -351,6 +352,7 @@ class ModelBuildRun:
         logger.info(
             f"Time taken: " + f"{humanize.precisedelta(datetime.timedelta(seconds=solver.solvetime), format='%0.4f')}"
         )
+        logger.info(subprocess.check_output(["nvidia-smi"], shell=False, stderr=subprocess.STDOUT))
 
 
 class GridBuilder: