Spaces:
Running
on
Zero
Running
on
Zero
skip cuda if not available
Browse files- app.py +7 -3
- requirements.txt +2 -1
app.py
CHANGED
@@ -4,9 +4,13 @@ import os, sys, importlib.util, re
|
|
4 |
import gradio as gr
|
5 |
from PIL import Image
|
6 |
import torch
|
7 |
-
|
8 |
-
|
9 |
-
print(f"
|
|
|
|
|
|
|
|
|
10 |
|
11 |
# βββ Monkey-patch mmdet to remove its mmcv-version assertion βββ
|
12 |
spec = importlib.util.find_spec('mmdet')
|
|
|
4 |
import gradio as gr
|
5 |
from PIL import Image
|
6 |
import torch
|
7 |
+
|
8 |
+
try:
|
9 |
+
print(f"CUDA available: {torch.cuda.is_available()}")
|
10 |
+
print(f"CUDA version: {torch.version.cuda}")
|
11 |
+
print(f"GPU device: {torch.cuda.get_device_name(0)}")
|
12 |
+
except:
|
13 |
+
print('CUDA is not available !')
|
14 |
|
15 |
# βββ Monkey-patch mmdet to remove its mmcv-version assertion βββ
|
16 |
spec = importlib.util.find_spec('mmdet')
|
requirements.txt
CHANGED
@@ -7,4 +7,5 @@ opencv-python
|
|
7 |
mmengine
|
8 |
mmdet
|
9 |
mmpose
|
10 |
-
gradio
|
|
|
|
7 |
mmengine
|
8 |
mmdet
|
9 |
mmpose
|
10 |
+
gradio
|
11 |
+
requests
|