ComfyUI生成图片报错: CUDA error: the launch timed out and was terminated

507 阅读2分钟

写在前面

在尝试comfyUI example“2_pass_txt2img”,总是在生成第二张图片的时候报错(通过减小图片长宽,steps都不行。可能是我电脑配置太低)如下:

RuntimeError: CUDA error: the launch timed out and was terminated Compile with TORCH_USE_CUDA_DSA to enable device-side assertions.

起初以为是因为自己电脑nvidia 驱动和cuda版本低导致,然后通过升级驱动和cuda,问题并没有解决;最后又尝试关闭win 系统 TDR(这个可以通过安装的Nsight Monitor操作),问题得以解决。网上有人说可以修改注册表数据,这种没有尝试,留给其他人尝试吧~~~

NVIDIA 驱动和CUDA更新

NVIDIA 驱动下载

www.nvidia.cn/Download/in…

CUDA工具包下载

最新版本下载:

developer.nvidia.com/cuda-downlo…

安装过程中如果遇到下面提示,不用太担心,Visual Studio主要是用来开发Windows平台的应用程序。

Installed:

  • Nsight Monitor

Not Installed:

  • Nsight for Visual Studio 2022

Reason: VS2022 was not found

  • Nsight for Visual Studio 2019

Reason: VS2019 was not found

  • Nsight for Visual Studio 2017

Reason: VS2017 was not found

  • Integrated Graphics Frame Debugger and Profiler

Reason: see developer.nvidia.com/nsight-vsto…

  • Integrated CUDA Profilers

Reason: see developer.nvidia.com/nsight-vsto…

其他版本下载

developer.nvidia.com/cuda-toolki…

CUDA-X

NVIDIA CUDA-X库基于 CUDA构建的一系列库,与仅使用 CPU 的替代方案相比,这些库在跨应用领域(包括 AI 和高性能计算)提供显着更高的性能。

www.nvidia.cn/technologie…

developer.nvidia.com/gpu-acceler…

NVIDIA DALI

NVIDIA 数据加载库 (DALI) 是一个便携式开源软件库,用于解码和增强图像、视频和语音,以加速深度学习应用程序。

developer.nvidia.com/DALI

cuDNN

NVIDIA cuDNN 是一个用于深度神经网络的 GPU 加速基元库

最新版本下载:developer.nvidia.com/cudnn

其他版本下载:developer.nvidia.com/rdp/cudnn-a…

关闭TRD

在安装CUDA工具的时候携带的Nsight Monitor工具,以管理员的方式打开,然后设置TRD为false。

效果

把steps调大了之后也能正常

问题之外

升级驱动和CUDA是否需要卸载老的版本

这个是没有必要的,升级的过程中,程序会自动删除老的应用,然后替换成新版本的东西。

还有更多其他问题,可以在这里查看:mp.weixin.qq.com/s/2_pEFIftr…