From ff0bf8c1d6023ce380bdfbb68f9870d81d0f68e2 Mon Sep 17 00:00:00 2001 From: project6-dev Date: Mon, 10 Aug 2026 06:21:40 +0000 Subject: [PATCH] =?UTF-8?q?fix:=20total=5Fmem=20=E2=86=92=20total=5Fmemory?= =?UTF-8?q?=20(torch=20API)?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- verify_single_gpu.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/verify_single_gpu.py b/verify_single_gpu.py index 72901544..9105eee7 100644 --- a/verify_single_gpu.py +++ b/verify_single_gpu.py @@ -356,7 +356,7 @@ def main(): print(f" CUDA available: {torch.cuda.is_available()}") if torch.cuda.is_available(): print(f" Device: {torch.cuda.get_device_name(0)}") - print(f" Memory: {torch.cuda.get_device_properties(0).total_mem / 1e9:.1f} GB") + print(f" Memory: {torch.cuda.get_device_properties(0).total_memory / 1e9:.1f} GB") print("=" * 60) results = {}