Ubuntu环境下启动ComfyUI时报错nVidia驱动过低问题解决

在Ubuntu容器内运行Comfy UI时报错:

[lolicp ComfyUI]$python3 main.py 
Warning, you are using an old pytorch version and some ckpt/pt files might be loaded unsafely. Upgrading to 2.4 or above is recommended.
Traceback (most recent call last):
  File "/home/lolicp/work/ComfyUI/main.py", line 137, in <module>
    import execution
  File "/home/lolicp/work/ComfyUI/execution.py", line 13, in <module>
    import nodes
  File "/home/lolicp/work/ComfyUI/nodes.py", line 22, in <module>
    import comfy.diffusers_load
  File "/home/lolicp/work/ComfyUI/comfy/diffusers_load.py", line 3, in <module>
    import comfy.sd
  File "/home/lolicp/work/ComfyUI/comfy/sd.py", line 7, in <module>
    from comfy import model_management
  File "/home/lolicp/work/ComfyUI/comfy/model_management.py", line 221, in <module>
    total_vram = get_total_memory(get_torch_device()) / (1024 * 1024)
  File "/home/lolicp/work/ComfyUI/comfy/model_management.py", line 172, in get_torch_device
    return torch.device(torch.cuda.current_device())
  File "/home/lolicp/anaconda3/envs/PyTorch-2.1.0/lib/python3.9/site-packages/torch/cuda/__init__.py", line 769, in current_device
    _lazy_init()
  File "/home/lolicp/anaconda3/envs/PyTorch-2.1.0/lib/python3.9/site-packages/torch/cuda/__init__.py", line 298, in _lazy_init
    torch._C._cuda_init()
RuntimeError: The NVIDIA driver on your system is too old (found version 11040). Please update your GPU driver by downloading and installing a new version from the URL: http://www.nvidia.com/Download/index.aspx Alternatively, go to: https://pytorch.org to install a PyTorch version that has been compiled with your version of the CUDA driver.

解决思路

查看nVidia驱动

发现cuda版本为11.4，与报错提示的11040一致。

[lolicp ComfyUI]$nvidia-smi 
Thu May  1 18:52:34 2025       
+-----------------------------------------------------------------------------+
| NVIDIA-SMI 470.57.02    Driver Version: 470.57.02    CUDA Version: 11.4     |
|-------------------------------+----------------------+----------------------+
---

查看python模块

查看torch模块版本，发现cuda使用的是12.1

[lolicp ComfyUI]$python3 -m pip list |grep torch
torch                          2.1.0+cu121
torchaudio                     2.1.0+cu121
torchvision                    0.16.0+cu121

卸载模块

pip3 uninstall -y torch torchaudio torchvision

安装模块

python3.9 -m pip install torch==1.12.1+cu113 torchaudio==0.12.1+cu113 torchvision==0.13.1+cu113 -i https://download.pytorch.org/whl/cu113

安装完成后再次运行即可。

END