fix: total_mem → total_memory (torch API)

2026-08-10 06:21:40 +00:00
parent c579c75039
commit ff0bf8c1d6
1 changed files with 1 additions and 1 deletions
--- 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 = {}