Stable Diffusion WebUI 在NVIDIA RTX 50系列显卡上的兼容性问题与解决方案

问题背景

近期，许多用户在NVIDIA最新发布的RTX 50系列显卡（如RTX 5070 Ti、RTX 5090等）上运行Stable Diffusion WebUI时遇到了兼容性问题。主要症状表现为启动时出现警告信息："NVIDIA GeForce RTX 5070 Ti with CUDA capability sm_120 is not compatible with the current PyTorch installation"，随后模型加载失败。

技术原因分析

这一问题的根本原因在于PyTorch官方发布的稳定版本尚未支持RTX 50系列显卡的SM_120架构。当前稳定版的PyTorch仅支持到SM_50至SM_90的CUDA计算能力。RTX 50系列采用了新一代的Blackwell架构，需要PyTorch的夜间构建版(nightly build)才能提供完整的支持。

详细解决方案

手动安装PyTorch夜间构建版

1. 首先需要卸载当前安装的PyTorch和TorchVision：

   python -m pip uninstall torch torchvision
   

2. 然后安装支持CUDA 12.8的夜间构建版：

   python -m pip install --pre torch torchvision --index-url https://download.pytorch.org/whl/nightly/cu128
   

注意事项

1. 安装过程需要下载约3GB的文件，对于网络条件不佳的用户可能需要较长时间。

2. 夜间构建版可能存在一些不稳定因素，建议定期更新以获取最新的修复和改进。

3. 对于Linux用户，特别是Ubuntu系统，目前官方尚未提供预编译的夜间构建版，可能需要从源代码编译。

性能表现

根据用户反馈，在成功配置后，RTX 50系列显卡在Stable Diffusion WebUI中表现出色：

• 高分辨率修复和放大功能运行速度显著提升
• 图像生成速度相比前代RTX 40系列有显著提高
• 能够流畅处理更高分辨率的图像生成任务

替代方案

对于不想使用夜间构建版的用户，可以考虑以下替代方案：

1. 使用Stable Diffusion的衍生版本，如Forge，这些项目可能已经整合了对新硬件的支持。

2. 等待PyTorch官方发布支持SM_120架构的稳定版本。

总结

虽然当前Stable Diffusion WebUI在RTX 50系列显卡上的支持还存在一些挑战，但通过安装PyTorch的夜间构建版可以解决主要的兼容性问题。随着时间推移和软件更新，这一问题有望得到更完善的解决。对于追求稳定性的用户，建议关注官方更新动态，在稳定版发布后再进行升级。