feat: 添加 Windows 兼容性支持,优化设备选择逻辑
- start_system.py: 跨平台进程管理 - Windows 使用 CREATE_NEW_PROCESS_GROUP 和 terminate() - Unix 使用 setsid 和 killpg - 自动检测并启用 Windows ANSI 颜色支持 - 设备选择逻辑优化: - macOS: 优先使用 MPS,回退到 CPU - Windows/Linux: 强制使用 CPU(无 CUDA 依赖) - 影响文件: - api/prediction_service.py - auto_trade_polymarket.py - training/config.py
This commit is contained in:
1 parent
8cdfb2a8e2
commit
233f0c5d10
4 files changed
+95
-30
No files matched your search
+11
-6
@@ -38,18 +38,23 @@ class TrainingConfig:
|
||||
|
||||
Returns:
|
||||
Device string ('mps', 'cuda', or 'cpu')
|
||||
|
||||
Note:
|
||||
macOS 使用 MPS,Windows/Linux 使用 CPU
|
||||
"""
|
||||
import torch
|
||||
import platform
|
||||
|
||||
if self.device != 'auto':
|
||||
return self.device
|
||||
|
||||
# Auto-detect best available device
|
||||
if torch.backends.mps.is_available():
|
||||
return 'mps'
|
||||
elif torch.cuda.is_available():
|
||||
return 'cuda'
|
||||
else:
|
||||
# 根据操作系统选择设备:macOS 用 MPS,Windows/Linux 用 CPU
|
||||
if platform.system() == 'Darwin': # macOS
|
||||
if torch.backends.mps.is_available():
|
||||
return 'mps'
|
||||
else:
|
||||
return 'cpu'
|
||||
else: # Windows / Linux
|
||||
return 'cpu'
|
||||
|
||||
|
||||
|
||||
Reference in new issue
Block a user