alrichardbollans commited on
Commit
f8af587
·
1 Parent(s): 0f2eb17

check install in app

Browse files
Files changed (1) hide show
  1. app.py +11 -0
app.py CHANGED
@@ -1,3 +1,14 @@
 
 
 
 
 
 
 
 
 
 
 
1
  import faicons as fa
2
  import plotly.express as px
3
 
 
1
+ import detectron2
2
+ import torch
3
+
4
+ # Check this in logs
5
+ try:
6
+ print(f"Is CUDA available: {torch.cuda.is_available()}")
7
+ # True
8
+ print(f"CUDA device: {torch.cuda.get_device_name(nn.cuda.current_device())}")
9
+ except:
10
+ print('Couldnt find CUDA device')
11
+
12
  import faicons as fa
13
  import plotly.express as px
14