Some debugging for CUDA mem leak

这个提交包含在:
craig-warren
2023-12-10 21:10:30 +00:00
父节点 7ac9f7f734
当前提交 38661d5868

查看文件

@@ -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: