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 = {}