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.
"""
if iteration == self.grid.iterations - 1:
mem_used = subprocess.check_output(["nvidia-smi", "--query-gpu=memory.used"],
shell=False, stderr=subprocess.STDOUT)
return mem_used
# # Total minus free memory in current context
# return self.drv.mem_get_info()[1] - self.drv.mem_get_info()[0]
# 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):
"""Calculates solving time for model."""