Update app.py

app.py

@@ -4,16 +4,11 @@ print('device:', device)
 torch_version = torch.__version__
 print('torch_version:', torch_version)
 
-# 获取服务器cuda版本
-def get_cuda_version():
-    try:
-        import torch
-        cuda_version = torch.version.cuda
-        if cuda_version is None:
-            return "CUDA version not found"
-        return cuda_version
-    except ImportError:
-        return "PyTorch is not installed"
-
-cuda_version = get_cuda_version()
-print('cuda_version:', cuda_version)
+# 获取机器的CUDA版本
+# 在机器中输入命令nvidia-smi
+import subprocess
+try:
+    result = subprocess.run(['nvidia-smi'], stdout=subprocess.PIPE, stderr=subprocess.PIPE, text=True)
+    print('nvidia-smi output:', result.stdout)
+except Exception as e:
+    print(f'Error: {e}')