ModuleNotFoundError: No module named ‘torch._dynamo’

问题场景

用户在 Kohya SS 中训练 LoRA，执行训练脚本时，程序在加载或初始化阶段抛出异常。日志显示已经通过 requirements 检查，开始加载模型和准备 accelerator，随后触发 torch._dynamo 导入错误。

报错原文

ModuleNotFoundError: No module named 'torch._dynamo'

import torch._dynamo as dynamo

原因分析

可能原因是 accelerate 配置中启用了 dynamo，但当前安装的 PyTorch 版本不包含 torch._dynamo 模块，或者 PyTorch 版本与 accelerate 的预期不兼容。torch._dynamo 是较新版本 PyTorch 才提供的模块，若环境中的 PyTorch 版本过旧或安装不完整，就会在导入时失败。

环境排查

• 确认 Python 版本，Kohya SS 不同版本对 Python 版本有不同要求。
• 确认 PyTorch 版本，Issue 中用户建议 torch==2.0.1，但未说明原始版本。
• 确认 CUDA 版本与 PyTorch 是否匹配。
• 确认 accelerate 版本，日志中显示使用 accelerator 0.15.0 或以上。
• 确认是否在 accelerate config 中启用了 dynamo。
• 确认是否使用 xformers、AdamW8bit 等依赖，但这些不是导致该报错的直接原因。

解决步骤

1. 重新运行 accelerate config，检查其中是否有类似“Do you want to use dynamo”或“Enable dynamo”的选项。
2. 在该选项处选择 No，然后保存配置。
3. 重新启动 Kohya SS 训练脚本，观察是否仍然出现 torch._dynamo 导入错误。
4. 如果问题仍存在，可优先尝试安装 PyTorch 2.0.1：pip install torch==2.0.1。该步骤来自评论中其他用户的反馈，未在原始 Issue 中完整验证。
5. 如果使用 venv，请确认安装命令作用于 Kohya SS 的虚拟环境，而不是系统 Python。

验证方法

重新运行训练命令，确认不再出现 ModuleNotFoundError: No module named ‘torch._dynamo’，并且训练流程能够继续进入模型加载或训练步骤。如果日志中不再有 dynamo 相关导入错误，说明问题已解决。

参考来源

bmaltais/kohya_ss #567

AI 工具推荐

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

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

了解 GamsGo AI

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