Update app.py
Browse files
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")
|