alok94 commited on
Commit
a6300b2
·
1 Parent(s): fc093ef
Files changed (1) hide show
  1. app.py +6 -0
app.py CHANGED
@@ -1,2 +1,8 @@
1
  import torch
2
  import gradio
 
 
 
 
 
 
 
1
  import torch
2
  import gradio
3
+
4
+ import torch
5
+ print(f"Is CUDA available: {torch.cuda.is_available()}")
6
+ # True
7
+ print(f"CUDA device: {torch.cuda.get_device_name(torch.cuda.current_device())}")
8
+ # Tesla T4