Skip to content

PaperMiner v1.4.22

Choose a tag to compare

@Given-Dream Given-Dream released this 05 Sep 04:50

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 的“重装”。