| """utils/device.py — Hardware device detection (CUDA / MPS / CPU)""" | |
| import sys | |
| def get_device() -> str: | |
| try: | |
| import torch | |
| if torch.cuda.is_available(): | |
| return "cuda" | |
| # Check MPS properly for Apple Silicon | |
| if hasattr(torch, "mps") and hasattr(torch.mps, "is_available"): | |
| if torch.mps.is_available(): | |
| return "mps" | |
| if hasattr(torch.backends, "mps") and torch.backends.mps.is_available(): | |
| return "mps" | |
| except ImportError: | |
| pass | |
| # Platform-based fallback: detect Apple Silicon without torch | |
| import platform, subprocess | |
| if platform.system() == "Darwin": | |
| try: | |
| result = subprocess.run(["sysctl", "-n", "hw.optional.arm64"], | |
| capture_output=True, text=True) | |
| if result.stdout.strip() == "1": | |
| return "mps" | |
| except Exception: | |
| pass | |
| return "cpu" | |
| def get_device_label() -> str: | |
| device = get_device() | |
| return {"cuda": "🟢 CUDA GPU", "mps": "🔵 Apple MPS", "cpu": "⚪ CPU"}.get(device, device) | |