new

app.py

@@ -56,6 +56,7 @@ def check_gpu():
     os.environ['LD_LIBRARY_PATH'] = "/usr/local/cuda-12.1/lib64:" + os.environ.get('LD_LIBRARY_PATH', '')
     subprocess.run(['nvidia-smi'])  # 测试 CUDA 是否可用
     print(f"torch.cuda.is_available:{torch.cuda.is_available()}")
+    print("Device count:", torch.cuda.device_count())
 check_gpu()