From 33a1a822e3abb771fa51ffb7171d76cd1e40b6cc Mon Sep 17 00:00:00 2001 From: Stas Bekman Date: Thu, 7 Mar 2024 14:04:08 -0800 Subject: [PATCH] benchmark header: add gpu info (#34) --- benchmarks/sizing/utils.py | 1 + 1 file changed, 1 insertion(+) diff --git a/benchmarks/sizing/utils.py b/benchmarks/sizing/utils.py index 59f85ce..f7325a0 100644 --- a/benchmarks/sizing/utils.py +++ b/benchmarks/sizing/utils.py @@ -23,6 +23,7 @@ def print_benchmark_header(notes="None"): ** Platform: {" ".join(platform.uname())} +{torch.cuda.get_device_properties(torch.device('cuda'))} ** Critical component versions: torch={torch.__version__}, cuda={torch.version.cuda}, nccl={torch.cuda.nccl.version()}