[bug]: Generation fails on ROCm 7.1 with torch 2.12.x (torch 2.11.0 works)

该报错发生在 AMD ROCm 7.1 环境下,当安装器默认解析到 PyTorch 2.12.x(含 2.13.0)时,InvokeAI 文生图会失败或导致显卡驱动重置。优先排查当前安装的 torch 版本,并固定为 torch 2.11.0 或升级到 ROCm 7.2 搭配 torch 2.13.

快速结论:该报错发生在 AMD ROCm 7.1 环境下,当安装器默认解析到 PyTorch 2.12.x(含 2.13.0)时,InvokeAI 文生图会失败或导致显卡驱动重置。优先排查当前安装的 torch 版本,并固定为 torch 2.11.0 或升级到 ROCm 7.2 搭配 torch 2.13.0/2.14.0 可恢复生成。

适用环境:已确认涉及 Linux 操作系统、AMD 显卡(RX 9070XT)、ROCm 7.1(部分用户使用 7.2)、InvokeAI 6.13.5rc1/6.13.6。所有证据均来自 ROCm 环境,未涉及 CUDA、CPU、Windows、ARM64 或 macOS MPS。

最快修复方案:在安装命令中将 torch 版本固定为 2.11.0(添加 "torch<2.12" 约束),或改用 ROCm 7.2 索引并升级至 torch 2.13.0 或 2.14.0——后者经两位用户确认可正常工作。

注意事项:2.11.0 仅为临时规避方案,底层缺陷尚未定位;Issue 关闭时没有任何 traceback 或最小复现,具体故障操作未知。ROCm 7.2 + torch 2.13.0/2.14.0 虽已验证可用,但该环境下 InvokeAI 仍无官方批量验证结论。

问题场景

用户在 Linux + AMD (ROCm) 系统上安装 InvokeAI,使用 --torch-backend=rocm7.1 让 torch 自由解析(当前默认得到 2.12.x)。运行 SDXL 批量生成 5 张图,在生成第二张图的 VAE 阶段触发图形子系统重置,导致 Invoke 启动器、Discord、Synology Drive 等桌面应用崩溃。另有用户报告在 torch 2.12.1 和 2.13.0 下模型加载耗时极长,而 2.11.0 只需数秒。

报错原文

[bug]: Generation fails on ROCm 7.1 with torch 2.12.x (torch 2.11.0 works)

原因分析

可能原因:InvokeAI 与 torch 2.12.x/2.13.0 在 ROCm 后端的交互存在兼容性问题。同一系统上 ComfyUI 搭配 ROCm 7.1 + torch 2.12.1 可正常运行,说明问题更可能出在 InvokeAI 使用 torch 的方式而非 ROCm wheel 本身。实际故障模式尚未确认——没有 traceback、没有失败算子、没有最小复现,只有“生成不可用”的现象描述。

环境排查

  • 操作系统:Linux(发行版未在 Issue 中细述)
  • 显卡:AMD RX 9070XT(来自原始报告者)
  • ROCm 版本:7.1(有用户改用 7.2)
  • InvokeAI 版本:6.13.5rc1 / 6.13.6
  • torch 问题版本:2.12.1、2.13.0;正常版本:2.11.0、2.13.0+rocm7.2、2.14.0+rocm7.2
  • Python 版本:3.12(来自安装命令)

解决步骤

  1. 确认当前安装的 torch 版本是否落在 2.12.x 或 2.13.0(rocm7.1 索引默认解析范围)。
  2. 可优先尝试:在安装命令中附加 "torch<2.12" 约束,使 torch 固定到 2.11.0,然后重新生成测试。该方案在 Issue 中经多位 AMD 用户验证有效。
  3. 另一可选路径:改用 ROCm 7.2 的 PyTorch wheel 索引,安装 torch 2.13.0+rocm7.2 或 2.14.0+rocm7.2。两位用户在各自系统上均确认这些版本工作正常,且 2.14.0 已解决先前遇到的故障。
  4. 若选择长期方案:等待 InvokeAI 官方修复 torch 2.12.x 兼容问题,或等待上游发布修复后的 torch/ROCm 版本。届时 Issue 关闭条件中包含移除手动安装文档中 torch<2.12 的警示。

验证方法

完成 torch 版本调整后,重新运行 SDXL 批量生成(至少覆盖 VAE 阶段),确认不再出现图形子系统重置或桌面应用崩溃;同时观察模型加载耗时是否恢复到正常秒级水平(对比 2.11.0 的表现)。Issue 关闭时维护者确认在测试 AMD 系统上 2.13.0+rocm7.2 与 2.14.0 均可正常生成。

参考来源

invoke-ai/InvokeAI #9410

GamsGo AI

AI 工具推荐

想把多个 AI 模型放在一个入口?

GamsGo AI 集成 ChatGPT、DeepSeek、Gemini、Claude、Midjourney、Veo 等常用模型,适合写作、绘图、视频和日常 AI 工作流。

了解 GamsGo AI

推广链接:通过此链接购买,我可能获得佣金,不影响你的价格。

这个方案解决了吗?

celebrityanime
celebrityanime
文章: 22353

发表回复

您的邮箱地址不会被公开。 必填项已用 * 标注