prithivMLmods commited on
Commit
4490767
·
verified ·
1 Parent(s): 83cef34

update app

Browse files
Files changed (1) hide show
  1. app.py +11 -0
app.py CHANGED
@@ -43,6 +43,17 @@ from gradio.themes.utils import colors, fonts, sizes
43
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
44
  print(f"✅ Using device: {device}")
45
 
 
 
 
 
 
 
 
 
 
 
 
46
  # --- Theme Definition ---
47
  colors.steel_blue = colors.Color(
48
  name="steel_blue",
 
43
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
44
  print(f"✅ Using device: {device}")
45
 
46
+ print("CUDA_VISIBLE_DEVICES=", os.environ.get("CUDA_VISIBLE_DEVICES"))
47
+ print("torch.__version__ =", torch.__version__)
48
+ print("torch.version.cuda =", torch.version.cuda)
49
+ print("cuda available:", torch.cuda.is_available())
50
+ print("cuda device count:", torch.cuda.device_count())
51
+ if torch.cuda.is_available():
52
+ print("current device:", torch.cuda.current_device())
53
+ print("device name:", torch.cuda.get_device_name(torch.cuda.current_device()))
54
+
55
+ print("Using device:", device)
56
+
57
  # --- Theme Definition ---
58
  colors.steel_blue = colors.Color(
59
  name="steel_blue",