developer-lunark commited on
Commit
a29ae54
ยท
verified ยท
1 Parent(s): da4532f

Show detailed GPU status in UI

Browse files
Files changed (1) hide show
  1. app.py +10 -1
app.py CHANGED
@@ -505,7 +505,16 @@ with gr.Blocks(title="KAIdol A/B Test Arena", theme=gr.themes.Soft()) as demo:
505
  gr.Markdown("# KAIdol A/B Test Arena")
506
  gr.Markdown("K-pop ์•„์ด๋Œ ๋กคํ”Œ๋ ˆ์ด ๋ชจ๋ธ A/B ๋น„๊ต ํ‰๊ฐ€ (์†Œํ˜• Student ๋ชจ๋ธ 11๊ฐœ)")
507
 
508
- mode_text = "**GPU ๋ชจ๋“œ**: ์‹ค์ œ ๋ชจ๋ธ ์ถ”๋ก " if GPU_AVAILABLE else "**Mock ๋ชจ๋“œ**: ํ…Œ์ŠคํŠธ ์‘๋‹ต ์ƒ์„ฑ"
 
 
 
 
 
 
 
 
 
509
  gr.Markdown(mode_text)
510
 
511
  with gr.Tabs():
 
505
  gr.Markdown("# KAIdol A/B Test Arena")
506
  gr.Markdown("K-pop ์•„์ด๋Œ ๋กคํ”Œ๋ ˆ์ด ๋ชจ๋ธ A/B ๋น„๊ต ํ‰๊ฐ€ (์†Œํ˜• Student ๋ชจ๋ธ 11๊ฐœ)")
507
 
508
+ # GPU ์ƒํƒœ ์ƒ์„ธ ์ •๋ณด
509
+ try:
510
+ import torch
511
+ torch_ver = torch.__version__
512
+ cuda_avail = torch.cuda.is_available()
513
+ cuda_ver = torch.version.cuda if cuda_avail else "N/A"
514
+ gpu_name = torch.cuda.get_device_name(0) if cuda_avail else "N/A"
515
+ mode_text = f"**GPU ๋ชจ๋“œ**: {gpu_name} (CUDA {cuda_ver}, PyTorch {torch_ver})" if cuda_avail else f"**Mock ๋ชจ๋“œ**: CUDA not available (PyTorch {torch_ver})"
516
+ except:
517
+ mode_text = "**Mock ๋ชจ๋“œ**: PyTorch not loaded"
518
  gr.Markdown(mode_text)
519
 
520
  with gr.Tabs():