Update app/routers/system.py
Browse files- app/routers/system.py +3 -1
app/routers/system.py
CHANGED
|
@@ -109,7 +109,9 @@ async def get_resource_usage():
|
|
| 109 |
disk = shutil.disk_usage('/')
|
| 110 |
|
| 111 |
# GPU info
|
| 112 |
-
|
|
|
|
|
|
|
| 113 |
|
| 114 |
if torch.cuda.is_available():
|
| 115 |
try:
|
|
|
|
| 109 |
disk = shutil.disk_usage('/')
|
| 110 |
|
| 111 |
# GPU info
|
| 112 |
+
|
| 113 |
+
gpu_available = torch.cuda.is_available()
|
| 114 |
+
gpu_info = GPUInfo(available=gpu_available, count=0, names=[])
|
| 115 |
|
| 116 |
if torch.cuda.is_available():
|
| 117 |
try:
|