运行AUTOMATIC1111 Stable Diffusion时出现CUDA内存不足问题

1. 启用WebUI专属低显存优化参数 #

启动WebUI时添加以下参数，优先推荐：

• --medvram：针对6GB显存卡优化，将模型分层加载到显存/内存，大幅降低显存占用
• --lowvram：比--medvram更激进，适合显存紧张场景
• --opt-split-attention：优化注意力层的显存分配，手动添加确保生效

示例启动命令（Linux环境）：

export PYTORCH_CUDA_ALLOC_CONF="garbage_collection_threshold:0.6,max_split_size_mb:512"
python launch.py --medvram --opt-split-attention

2. 调整生成参数减少显存占用 #

• 降低生成分辨率：优先使用512x512或更低尺寸，若需要高分辨率，用「高分辨率修复」功能先低分辨率生成再放大，避免直接生成大尺寸图
• 强制Batch Size为1：在WebUI生成界面将Batch size设为1，避免批量生成占用额外显存
• 关闭非必要功能：生成时暂时禁用面部修复（CodeFormer/GFPGAN）、超分辨率模型（ESRGAN/RealESRGAN），这类功能会额外占用数百MB显存

3. 优化PyTorch显存配置（补充已尝试的设置） #

在已设置max_split_size_mb=512的基础上，添加垃圾回收阈值配置，让PyTorch更及时释放闲置显存：
启动WebUI前设置环境变量：

export PYTORCH_CUDA_ALLOC_CONF="garbage_collection_threshold:0.6,max_split_size_mb:512"

该配置会让PyTorch在显存占用达到60%时触发垃圾回收，减少显存碎片问题

4. 系统层面释放显存（针对非Windows环境） #

• 关闭桌面环境硬件加速：GNOME/KDE等桌面环境的硬件加速会占用数十到数百MB显存，可改用轻量桌面（如Xfce）或在系统设置中关闭
• 结束后台GPU占用程序：关闭浏览器（尤其是带硬件加速的标签页）、视频播放器、其他AI工具等占用GPU的进程

内容的提问来源于stack exchange，提问作者Leirbag Ed Satierf