# -*- coding: utf-8 -*-
"""
简单GPU检测脚本 - 检测系统是否有显卡及数量
"""

import torch
import cv2

def check_gpu():
  """检测GPU数量"""
  print("🔍 正在检测GPU...")
  
  # 检查CUDA是否可用
  if torch.cuda.is_available():
      gpu_count = torch.cuda.device_count()
      print(f"✅ 检测到 {gpu_count} 张GPU")

      print(f"cv2 检测到{cv2.cuda.getCudaEnabledDeviceCount()}张gpu")
      
      # 显示GPU名称
      for i in range(gpu_count):
          gpu_name = torch.cuda.get_device_name(i)
          print(f"   GPU {i}: {gpu_name}")
  else:
      print("❌ 未检测到可用的GPU")
  
  # 检查MPS (Apple Silicon)
  if hasattr(torch.backends, 'mps') and torch.backends.mps.is_available():
      print("✅ 检测到Apple Silicon GPU (MPS)")

if __name__ == "__main__":
  check_gpu()
