mamkkl commited on
Commit
84cca03
·
verified ·
1 Parent(s): f2a1799

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -2
app.py CHANGED
@@ -67,8 +67,6 @@ def print_resources():
67
  print(f"GPU {i}: {torch.cuda.get_device_name(i)}")
68
  print(f" Total Memory: {torch.cuda.get_device_properties(i).total_memory / 1e9:.2f} GB")
69
  print(f" CUDA Capability: {torch.cuda.get_device_properties(i).major}.{torch.cuda.get_device_properties(i).minor}")
70
- for i in range(num_gpus):
71
- print(f"GPU {i}: {torch.cuda.get_device_name(i)}")
72
  print(f" Allocated Memory: {torch.cuda.memory_allocated(i) / 1e9:.2f} GB")
73
  print(f" Cached Memory: {torch.cuda.memory_reserved(i) / 1e9:.2f} GB")
74
  print(f" Free Memory: {torch.cuda.get_device_properties(i).total_memory / 1e9 - torch.cuda.memory_reserved(i) / 1e9:.2f} GB")
 
67
  print(f"GPU {i}: {torch.cuda.get_device_name(i)}")
68
  print(f" Total Memory: {torch.cuda.get_device_properties(i).total_memory / 1e9:.2f} GB")
69
  print(f" CUDA Capability: {torch.cuda.get_device_properties(i).major}.{torch.cuda.get_device_properties(i).minor}")
 
 
70
  print(f" Allocated Memory: {torch.cuda.memory_allocated(i) / 1e9:.2f} GB")
71
  print(f" Cached Memory: {torch.cuda.memory_reserved(i) / 1e9:.2f} GB")
72
  print(f" Free Memory: {torch.cuda.get_device_properties(i).total_memory / 1e9 - torch.cuda.memory_reserved(i) / 1e9:.2f} GB")