sunana commited on
Commit
3ffe73b
1 Parent(s): d09cb1e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -1
app.py CHANGED
@@ -7,7 +7,12 @@ import sys
7
  import FFV1MT_MS
8
  import flow_tools
9
 
10
-
 
 
 
 
 
11
  model = FFV1MT_MS.FFV1DNN()
12
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
13
 
 
7
  import FFV1MT_MS
8
  import flow_tools
9
 
10
+
11
+ print(f"Is CUDA available: {torch.cuda.is_available()}")
12
+ # True
13
+ print(f"CUDA device: {torch.cuda.get_device_name(torch.cuda.current_device())}")
14
+ # Tesla T4
15
+
16
  model = FFV1MT_MS.FFV1DNN()
17
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
18