xFormers 50 系列显卡兼容性问题
作者: KATe-AGI创建于 2025年10月5日更新于 2026年4月25日
问题 你好,我的环境配备了 PyTorch 2.8 和 CUDA 12.8。当我使用 50 系列显卡运行 pip3 install -U xformers --index-url https://download.pytorch.org/whl/cu128 时, xFormers 可以成功编译和安装。然而,在实际训练过程中, xFormers 无法正常运行,错误消息显示问题在于内核架构不兼容。我怀疑如果这个预编译包仅针对 H 系列显卡使用?
内容来源: facebookresearch/xformers