Cuda 12.9 and Pytorch 2.8 support

用户在安装或运行 xFormers(通常用于 ComfyUI 等工具)时,希望获取支持 CUDA 12.9 和 PyTorch 2.8 的预编译 wheel 包。常见于 Windows/Linux 环境。

Cuda 12.9 and Pytorch 2.8 support

Cuda 12.9 and Pytorch 2.8 support

快速结论:此报错发生在尝试安装或使用 xFormers 时遇到 CUDA 12.9 与 PyTorch 2.8 的组合。优先检查是否必须使用该旧版组合,如非必需,建议升级到 PyTorch 2.10+。

问题场景

用户在安装或运行 xFormers(通常用于 ComfyUI 等工具)时,希望获取支持 CUDA 12.9 和 PyTorch 2.8 的预编译 wheel 包。常见于 Windows/Linux 环境。

报错原文

# ❓Cuda 12.9 and Pytorch 2.8 support

Is there any build that support cu129 and torch2.8?

原因分析

官方确认不会为 PyTorch 2.8 + CUDA 12.9 发布新的 xFormers 构建。原因如下:

  • PyTorch 2.8 与 CUDA 12.9 是已被后续版本取代的旧版本。
  • 从 PyTorch 2.10 开始,xFormers 对不同 PyTorch 和 CUDA 版本的兼容性显著改善。
  • 官方推荐用户升级到 PyTorch 2.10+(及对应 CUDA 版本),以获得更好的支持。

环境排查

  • 当前 PyTorch 版本:检查是否确实为 2.8.x
  • 当前 CUDA 版本:nvidia-smi 输出
  • 操作系统:Windows / Linux
  • Python 版本:通常为 3.12

解决步骤

  1. (可优先尝试)升级 PyTorch 和 CUDA:将 PyTorch 升级到 2.10+,并将 CUDA 升级到对应支持的版本(如 CUDA 12.4 或更高),官方建议此方案。
  2. 如果必须保留 PyTorch 2.8 + CUDA 12.9:需要从源码构建 xFormers:
    • 克隆 xFormers 仓库
    • 根据仓库 README 中的指导执行源码编译(注意:此过程可能较复杂且需自行调试兼容问题)
  3. 对于 Windows Python 3.12 用户(如 ComfyUI 用户):目前官方未提供对应 wheel,只能源码编译,或等社区第三方构建(无官方支持)。

验证方法

成功安装后,运行以下命令确认 xFormers 可正常加载且无 CUDA 相关错误:

import xformers
print(xformers.__version__)
# 并尝试运行一个使用 xFormers 的简单模型

同时检查一切依赖版本是否匹配。

参考来源

facebookresearch/xformers #1261

GamsGo AI

AI 工具推荐

想把多个 AI 模型放在一个入口?

GamsGo AI 集成 ChatGPT、DeepSeek、Gemini、Claude、Midjourney、Veo 等常用模型,适合写作、绘图、视频和日常 AI 工作流。

了解 GamsGo AI

推广链接:通过此链接购买,我可能获得佣金,不影响你的价格。

celebrityanime
celebrityanime
文章: 7742

发表回复

您的邮箱地址不会被公开。 必填项已用 * 标注