benchmark header: add gpu info (#34)

EleutherAI · Mar 7, 2024 · 33a1a82 · 33a1a82
1 parent 99ecff6
commit 33a1a82
Showing 1 changed file with 1 addition and 0 deletions.
diff --git 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()}