diff --git a/gprMax/model_build_run.py b/gprMax/model_build_run.py
index 397682c5..93e812bd 100644
--- a/gprMax/model_build_run.py
+++ b/gprMax/model_build_run.py
@@ -352,7 +352,6 @@ 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:
diff --git a/gprMax/updates.py b/gprMax/updates.py
index 834017ec..b2e93698 100644
--- a/gprMax/updates.py
+++ b/gprMax/updates.py
@@ -235,8 +235,6 @@ class CUDAUpdates:
 
         # Import PyCUDA modules
         self.drv = import_module("pycuda.driver")
-        self.clear_context_caches = getattr(import_module("pycuda.tools"), 
-                                            "clear_context_caches")
         self.source_module = getattr(import_module("pycuda.compiler"), "SourceModule")
         self.drv.init()
 
@@ -786,11 +784,9 @@ class CUDAUpdates:
 
     def cleanup(self):
         """Cleanup GPU context."""
-        # Remove context from top of stack and delete
+        # Remove context from top of stack and clear
         self.ctx.pop()
         self.ctx = None
-        self.clear_context_caches()
-
 
 
 class OpenCLUpdates: