33 lines
837 B
Python
33 lines
837 B
Python
# -*- 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()
|