Revert mem use estimate

这个提交包含在:
craig-warren
2023-12-10 23:07:24 +00:00
父节点 95853088d9
当前提交 ef5532860a

查看文件

@@ -754,11 +754,8 @@ class CUDAUpdates:
Memory (RAM) used on GPU. Memory (RAM) used on GPU.
""" """
if iteration == self.grid.iterations - 1: if iteration == self.grid.iterations - 1:
mem_used = subprocess.check_output(["nvidia-smi", "--query-gpu=memory.used"], # Total minus free memory in current context
shell=False, stderr=subprocess.STDOUT) return self.drv.mem_get_info()[1] - self.drv.mem_get_info()[0]
return mem_used
# # Total minus free memory in current context
# return self.drv.mem_get_info()[1] - self.drv.mem_get_info()[0]
def calculate_solve_time(self): def calculate_solve_time(self):
"""Calculates solving time for model.""" """Calculates solving time for model."""