Stable Diffusion扩展SadTalker仅用CPU运行，GPU未利用求助

针对AMD RX 580 8GB显卡下SadTalker不调用GPU、但txt2img功能正常的情况，可尝试以下方案：

1. 手动切换SadTalker的GPU配置 #

打开Stable Diffusion WebUI的SadTalker扩展设置页，找到GPU设备选择或启用GPU加速相关选项，确保已选中对应的AMD GPU设备，而非默认的CPU选项。部分版本的SadTalker会默认优先使用CPU，需手动切换。

2. 补充AMD专属启动参数 #

在webui-user.bat的启动参数中添加AMD显卡适配参数：

• 追加--rocm（适配AMD的ROCm计算框架，需确保已安装对应驱动）
• 若为Windows系统，可额外添加--no-half-vae，避免显存不足导致GPU fallback到CPU

修改后的webui-user.bat启动参数示例：

set COMMANDLINE_ARGS=--medvram --disable-safe-unpickle --rocm --no-half-vae

3. 验证PyTorch对AMD GPU的识别 #

打开WebUI的终端窗口，执行以下代码检查PyTorch是否正确识别GPU：

import torch; print(torch.cuda.is_available())

若返回True，说明PyTorch已适配GPU；若返回False，需重新安装适配AMD的ROCm版本PyTorch。

4. 更新SadTalker扩展 #

在WebUI的扩展管理页面，找到SadTalker并点击更新按钮。部分旧版本的SadTalker对AMD显卡支持不完善，更新后可修复GPU调用逻辑。

5. 调整显存分配策略 #

若--medvram参数仍无法满足SadTalker的显存需求，可尝试：

• 替换为--lowvram参数进一步降低显存占用
• 暂时移除--medvram，使用默认显存模式（若系统剩余显存充足）

内容的提问来源于stack exchange，提问作者Katy Brown