vectorplasticity commited on
Commit
ac63c9d
·
verified ·
1 Parent(s): ecc1fa7

Update app/routers/system.py

Browse files
Files changed (1) hide show
  1. 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
- gpu_info = GPUInfo(available=False, count=0, names=[])
 
 
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: