Show detailed GPU status in UI
Browse files
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 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 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():
|