ModuleNotFoundError: No module named ‘comfy_aimdo.vram_buffer’

快速结论：此报错通常发生在 AMD GPU（非 NVIDIA）用户启动 ComfyUI 时，由于缺少 NVIDIA CUDA 库导致 `comfy_aimdo` 模块加载失败。优先排查是否安装并更新了 `comfy_aimdo` 包，或已在 `model_management.py` 中添加兼容性处理。

问题场景

用户在 Ubuntu Linux（AMD GPU）系统上启动 ComfyUI 时触发。ComfyUI 在初始化过程中尝试导入 `comfy_aimdo.vram_buffer`，但该模块因缺少 NVIDIA CUDA 运行时库（`libcuda.so.1`）而无法加载，最终导致 ComfyUI 启动进程以退出码 1 终止。

报错原文

setup plugin alembic.autogenerate.schemas
setup plugin alembic.autogenerate.tables
...
comfy-aimdo failed to load: libcuda.so.1: cannot open shared object file: No such file or directory
NOTE: comfy-aimdo is currently only support for Nvidia GPUs
Traceback (most recent call last):
  File "/mnt/data/code/ComfyUI/main.py", line 198, in <module>
    import execution
  File "/mnt/data/code/ComfyUI/execution.py", line 17, in <module>
    import comfy.model_management
  File "/mnt/data/code/ComfyUI/comfy/model_management.py", line 34, in <module>
    import comfy_aimdo.vram_buffer
ModuleNotFoundError: No module named 'comfy_aimdo.vram_buffer'

原因分析

可能原因：`comfy_aimdo` 包在非 NVIDIA GPU（如 AMD）环境下仅支持 NVIDIA，且可能因版本过旧或未实现 AMD 兼容性导致 `comfy_aimdo.vram_buffer` 子模块未被正确安装或导入。Issue 评论指出该问题在最近更新中突然出现，需要用户手动修补或升级 `comfy_aimdo`。

环境排查

确认 Python 版本：用户运行环境为 Python 3.14.4
确认操作系统：Linux（Ubuntu latest amd）
确认显卡类型：AMD GPU（非 NVIDIA），用户明确使用了 AMD 平台
确认 ComfyUI Manager 配置：在 ComfyUI 用户目录 user/__manager/config.ini
确认 comfy_aimdo 版本：未明确版本号，但评论建议升级 comfy_aimdo

解决步骤

打开 comfy/model_management.py 文件，定位到第 34 行附近，找到 import comfy_aimdo.vram_buffer 语句。

将原有导入行注释掉，并在下方添加兼容性导入代码：

try:
    import comfy_aimdo.vram_buffer
except (ModuleNotFoundError, ImportError):
    comfy_aimdo = None

可优先尝试：在 ComfyUI 的 Python 环境中升级 comfy_aimdo 包。Windows 用户在虚拟环境中执行命令 .\.venv\Scripts\python.exe -m pip install comfy_aimdo -U，Linux 用户使用 python -m pip install comfy_aimdo -U（确保命令在 ComfyUI 的 Python 目录下运行，若使用虚拟环境 venv 则需指定 venv 内的 Python 解释器路径）。
保存更改并重新启动 ComfyUI。

验证方法

重新启动 ComfyUI 后，观察终端输出或日志文件中是否不再出现 ModuleNotFoundError: No module named 'comfy_aimdo.vram_buffer' 错误，且界面正常加载。确认 model_management.py 中的 try-except 块未引发未捕获异常，或终端成功输出 comfy-aimdo failed to load 的非致命提示后继续执行。