From c6b9ee93e942bf31cc6ceef25668ddc089930389 Mon Sep 17 00:00:00 2001 From: Claude Date: Tue, 11 Aug 2026 13:18:53 +0000 Subject: [PATCH] =?UTF-8?q?fix:=20total=5Fmem=20=E2=86=92=20total=5Fmemory?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- verify_forward.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/verify_forward.py b/verify_forward.py index b97682f5..465e3b86 100644 --- a/verify_forward.py +++ b/verify_forward.py @@ -11,7 +11,7 @@ os.environ.setdefault("BI100_GDN_COREX_PACKED_DECODE", "1") import torch print(f"torch {torch.__version__}, CUDA {torch.cuda.is_available()}") if torch.cuda.is_available(): - print(f"GPU: {torch.cuda.get_device_name(0)}, {torch.cuda.get_device_properties(0).total_mem // 1024**2} MB") + print(f"GPU: {torch.cuda.get_device_name(0)}, {torch.cuda.get_device_properties(0).total_memory // 1024**2} MB") # Step 1: 验证所有.so加载 print("\n=== Step 1: .so加载 ===")