Fooocus项目在AMD GPU上的Segfault问题分析与解决方案

问题背景

Fooocus作为一款基于深度学习的图像生成工具，在AMD GPU平台上运行时经常遇到Segmentation Fault(段错误)问题。这一问题主要影响使用Radeon RX 6000系列显卡的Linux用户，表现为在图像生成过程中程序意外崩溃。

问题现象

用户报告的主要症状包括：

1. 程序在初始化阶段或图像生成过程中突然崩溃
2. 控制台输出"Segmentation fault (core dumped)"错误信息
3. 部分用户观察到内存访问错误提示："Memory access fault by GPU node-1"
4. 使用--always-cpu参数可以运行但性能极低

根本原因分析

经过社区多方验证，这一问题主要源于AMD ROCm与PyTorch的兼容性问题，具体表现为：

1. GPU架构识别错误：AMD Radeon RX 6000系列(GFX10.3架构)被错误识别为其他架构版本
2. 内存管理异常：GPU驱动无法正确处理显存分配请求
3. ROCm版本不匹配：部分系统安装的ROCm运行时与PyTorch预期版本不一致

解决方案

针对不同型号的AMD GPU，社区发现了有效的解决方案：

对于Radeon RX 6000系列(如6700 XT、6600 XT等)

HSA_OVERRIDE_GFX_VERSION=10.3.0 python launch.py

对于Radeon RX 7000系列(如7800 XT等)

HSA_OVERRIDE_GFX_VERSION=11.0.0 python launch.py

通用优化建议

1. 确保ROCm环境正确安装：

   • 安装rocm-opencl和rocm-hip-runtime软件包
   • 验证ROCm版本与PyTorch要求的版本匹配

2. 使用正确的PyTorch版本：

   pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/rocm5.6
   

3. 内存优化参数：

   • 对于显存较小的显卡，可添加--always-low-vram参数
   • 大内存系统可配置充足的swap空间(建议40GB以上)

技术原理

HSA_OVERRIDE_GFX_VERSION环境变量强制指定了GPU的架构版本，解决了PyTorch错误识别AMD GPU架构的问题。这一变量直接影响ROCm运行时如何与特定GPU型号交互，确保正确的内存分配和计算内核选择。

验证结果

社区反馈表明：

• RX 6700 XT用户使用10.3.0参数后问题完全解决
• RX 7800 XT用户使用11.0.0参数后运行正常
• 性能恢复到预期水平，无稳定性问题

结论

AMD GPU用户在Fooocus项目中遇到的Segfault问题主要源于架构识别和ROCm兼容性问题。通过正确设置HSA_OVERRIDE_GFX_VERSION环境变量，配合适当的PyTorch版本，可以稳定高效地运行Fooocus图像生成任务。这一解决方案不仅适用于Fooocus，也可为其他基于PyTorch的AI应用在AMD平台上的部署提供参考。