Repository navigation
PaperMiner v1.4.22
PaperMiner v1.4.22
本版修复 NVIDIA RTX 30、40、50 系显卡的 PyTorch CUDA wheel 选择与自动修复逻辑。
主要变化
- 同时读取 GPU 型号、计算能力和 Windows 驱动版本,不再只按驱动号选择 CUDA。
- RTX 30 / Ampere、RTX 40 / Ada 根据驱动使用
cu126、cu121或cu118。 - RTX 50 / Blackwell 在 Windows 驱动 580.88 及更新版本上使用
cu130。 - RTX 50 在 570.65–580.87 驱动区间使用固定的 PyTorch 2.11.0
cu128兼容组合;更旧驱动提示升级并暂时使用 CPU wheel。 - 混合多卡按要求最高的一张选择统一 wheel,并在每张可见 GPU 上执行真实 FP32→FP16 运算与同步。
- 发现 CUDA runtime 不符、缺少当前 GPU kernel image 或内核运算失败时,Setup 会显示现有环境与目标方案,卸载旧
torch/torchvision/torchaudio后重装。 - PaperMiner 主程序的 GPU 预检也执行真实 kernel 测试;若仍发生
no kernel image,日志会明确提示使用 Setup 的“重装”。