Stable Diffusion WebUI AMDGPU版本中ROCm版本检测问题分析

问题背景

在Windows 10系统上使用Stable Diffusion WebUI AMDGPU版本时，当安装了HIP 6.1.2 SDK后，部分用户遇到了版本检测失败的问题。这个问题主要出现在1.10.0版本中，而之前的1.9.3-AMD版本则能正常工作。

问题现象

当用户使用--use-zluda参数启动WebUI时，系统会调用rocm.py脚本来检测HIP SDK版本。在特定配置环境下，脚本会返回空版本号，导致ZLUDA安装失败，最终只能回退到CPU模式运行。

技术分析

版本检测机制

rocm.py脚本主要通过两种方式获取HIP SDK路径：

1. 通过环境变量查找：首先尝试使用shutil.which("hipconfig")查找可执行文件路径
2. 通过HIP_PATH环境变量：当第一种方式失败时，尝试读取HIP_PATH环境变量

问题根源

当系统未配置%HIP_PATH%bin环境变量时，脚本会直接返回HIP_PATH环境变量的值（如C:\Program Files\AMD\ROCm\6.1\），但由于路径末尾包含反斜杠，导致后续版本号提取失败。

解决方案对比

1. 1.9.3-AMD版本：工作正常，说明其版本检测逻辑能够正确处理路径格式
2. 1.10.0版本：存在路径处理缺陷，导致版本号提取失败

临时解决方案

用户可以手动修改rocm.py脚本中的路径处理逻辑：

# 修改前
return hip_path

# 修改后
return dirname(resolve_link(hip_path), 1)

这个修改确保返回的路径格式统一，不包含末尾的反斜杠，使版本号提取能够正常工作。

版本兼容性建议

值得注意的是，不应将使用HIP SDK 5.7构建的旧版ROCmLibs与HIP 6.1混用，这可能导致不可预见的兼容性问题。用户应确保使用匹配版本的组件。

结论

这个问题主要源于路径字符串处理的边界情况未充分考虑。对于开发者而言，在路径处理时应特别注意：

• 路径分隔符的统一处理
• 路径末尾可能存在的分隔符
• 跨平台路径处理的兼容性

对于终端用户，建议等待官方修复或按照上述临时方案修改脚本，同时注意保持各组件版本的一致性。