jssck commited on
Commit
800276c
·
verified ·
1 Parent(s): 8c56f29

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -0
app.py CHANGED
@@ -164,3 +164,9 @@ with gr.Blocks() as demo:
164
 
165
  demo.queue(max_size=10)
166
  demo.launch()
 
 
 
 
 
 
 
164
 
165
  demo.queue(max_size=10)
166
  demo.launch()
167
+ import torch
168
+
169
+ if torch.cuda.is_available():
170
+ print(f"GPU disponible : {torch.cuda.get_device_name(0)}")
171
+ else:
172
+ print("Aucun GPU disponible")