prithivMLmods commited on
Commit
c7228f8
·
verified ·
1 Parent(s): a4e24f3

update app

Browse files
Files changed (1) hide show
  1. app.py +7 -0
app.py CHANGED
@@ -18,6 +18,13 @@ device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
18
 
19
  print("CUDA_VISIBLE_DEVICES=", os.environ.get("CUDA_VISIBLE_DEVICES"))
20
  print("torch.__version__ =", torch.__version__)
 
 
 
 
 
 
 
21
  print("Using device:", device)
22
 
23
  from diffusers import FlowMatchEulerDiscreteScheduler
 
18
 
19
  print("CUDA_VISIBLE_DEVICES=", os.environ.get("CUDA_VISIBLE_DEVICES"))
20
  print("torch.__version__ =", torch.__version__)
21
+ print("torch.version.cuda =", torch.version.cuda)
22
+ print("cuda available:", torch.cuda.is_available())
23
+ print("cuda device count:", torch.cuda.device_count())
24
+ if torch.cuda.is_available():
25
+ print("current device:", torch.cuda.current_device())
26
+ print("device name:", torch.cuda.get_device_name(torch.cuda.current_device()))
27
+
28
  print("Using device:", device)
29
 
30
  from diffusers import FlowMatchEulerDiscreteScheduler