m-ric HF staff commited on
Commit
2d63e52
1 Parent(s): 9a4471c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -0
app.py CHANGED
@@ -1,3 +1,8 @@
 
 
 
 
 
1
  STYLE = """
2
  .container {
3
  width: 100%;
 
1
+ import torch
2
+ print(f"Is CUDA available: {torch.cuda.is_available()}")
3
+ # True
4
+ print(f"CUDA device: {torch.cuda.get_device_name(torch.cuda.current_device())}")
5
+
6
  STYLE = """
7
  .container {
8
  width: 100%;