【Windows RTX 50XX显卡如何使用xformers加速和解决triton导入报错-linux版本运行会报其他错误】
一.下载官方的整合包
https://github.com/comfyanonymous/ComfyUI/releases
下载ComfyUI_windows_portable_nvidia_cu128.7z版本
这个版本对应的python版本是3.12.10, python版本过高可能会导致xFormers安装失败, 不建议使用桌面版,python版本不好控制,3.13版本安装windows的xFormer会报错,linux虽然不会报错, 但是运行时候又会报错,简直崩溃啊
二.xFormers兼容的torch版本[windows可用]
Python: 3.12.10
torch: 2.7.0+cu128
torchaudio: 2.7.0+cu128
torchvision: 0.22.0+cu128
torchsde: 0.2.6 # 好像非必须
xformers version: 0.0.30
安装命令
# 卸载,非必须
python -m pip uninstall torch torchvision torchaudio xformers -y
python -m pip install torch==2.7.0 torchvision==0.22.0 torchaudio==2.7.0 --index-url https://download.pytorch.org/whl/cu128
# 如果使用的python版本过高会报错, 我使用过3.13.*报错降了版本成功,网上有3.11.8可以的
python -m pip install xformers==0.0.30 --index-url https://download.pytorch.org/whl/cu128
如果无法下载可以找相关的轮子安装
https://download.pytorch.org/whl/cu128/
https://download.pytorch.org/whl/cu128/xformers/
python可以配置环境变量PIP_INDEX_URL=https://pypi.tuna.tsinghua.edu.cn/simple修改镜像地址
三.import triton错误解决
这个是最头大的问题,不影响使用,但是感觉不完美,https://pypi.org/网址只有linux的版本,官网提供的轮子也只有linux版本
本来想通过源码编译一个版本出来的,但是没有搞过搞着搞着有点小崩溃,突然翻到了一个轮子地址
https://github.com/woct0rdho/triton-windows/releases/v3.0.0-windows.post1/
下载triton-3.0.0-cp312-cp312-win_amd64.whl版本安装就解决了
欢迎来到FlagOS开发社区,这里是一个汇聚了AI开发者、数据科学家、机器学习爱好者以及业界专家的活力平台。我们致力于成为业内领先的Triton技术交流与应用分享的殿堂,为推动人工智能技术的普及与深化应用贡献力量。
更多推荐



所有评论(0)