Update app.py

app.py

@@ -1,3 +1,15 @@
+import torch
+import subprocess
+
+print("torch.cuda.is_available():", torch.cuda.is_available())
+try:
+    # 直接调用 nvidia-smi 查看驱动版本
+    nv_out = subprocess.check_output(["nvidia-smi", "--query-gpu=name,driver_version,cuda_version", "--format=csv,noheader"])
+    print(nv_out.decode("utf-8"))
+except Exception as e:
+    print("nvidia-smi 无法执行或不存在：", e)
+
+"""
 import os
 import torch
 from threading import Thread
@@ -485,4 +497,5 @@ with gr.Blocks() as demo:
                                   show_progress=True)
     
 
-demo.launch()
+demo.launch()
+"""