dylanyunlong/project_6_89d52222
1
0
Fork 0
Code Issues Pull Requests Actions Projects Releases Wiki Activity

fix: total_mem → total_memory (torch API)

Browse Source
This commit is contained in:
project6-dev
2026-08-10 06:21:40 +00:00
parent c579c75039
commit ff0bf8c1d6
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
verify_single_gpu.py
View File

@@ -356,7 +356,7 @@ def main():
print(f" CUDA available: {torch.cuda.is_available()}") print(f" CUDA available: {torch.cuda.is_available()}")
if torch.cuda.is_available(): if torch.cuda.is_available():
print(f" Device: {torch.cuda.get_device_name(0)}") 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) print("=" * 60)
results = {} results = {}