Update app.py

app.py

@@ -5,6 +5,20 @@ import numpy as np
 
 # 你已有的推理代码
 from inference import load_model, get_embedding, run
+import torch
+import os
+
+print("=== [DEBUG] Torch Info ===")
+print("Torch version:", torch.__version__)
+print("CUDA available:", torch.cuda.is_available())
+if torch.cuda.is_available():
+    print("CUDA device count:", torch.cuda.device_count())
+    print("CUDA device name:", torch.cuda.get_device_name(0))
+    print("CUDA version (torch):", torch.version.cuda)
+    print("cuDNN version:", torch.backends.cudnn.version())
+else:
+    print("Running on CPU only")
+print("==========================")
 
 # ---- 仅加载一次模型 ----
 model, device = load_model("medsam_vit_b.pth")