在Windows11下训练Stable-Diffusion的LoRA模型的时候,总是重复提示:
A matching Triton is not available, some optimizations will not be enabled.
Error caught was: No module named ‘triton’
意思是:
没有匹配的 Triton,一些优化将不会被启用。
捕获的错误是: 没有名为 "TRITON "的模块
虽然过程并不会中断,但是看着好难受。
上网搜了一下,应该是 xformers 需要用到 Triton 进行优化工作。
网上有人说:
并没有Windows下的Triton,所以Windows下会报错,而Linux就有Triton。
直接安装果然没有:
> pip install triton
ERROR: Could not find a version that satisfies the requirement triton (from versions: none)
ERROR: No matching distribution found for triton
看了下https://github.com/openai/triton,https://triton-lang.org/,https://pypi.org/project/triton/,
果然哪里都没有提供Windows的二进制下载,难道要自己编译?
但实际上有大神编译了Windows下二进制:
triton-2.0.0-cp310-cp310-win_amd64.whl
下载triton-2.0.0-cp310-cp310-win_amd64.whl
后安装:
> python.exe -m pip install triton-2.0.0-cp310-cp310-win_amd64.whl
结果又又提示要CMake……且自动下载cmake-3.26.3-py2.py3-none-win_amd64.whl
巨慢无比,天朝网络啊……
只好用迅雷下载,参考页面:https://pypi.org/project/cmake/#files。
然后手动安装。
> python.exe -m pip install cmake-3.26.3-py2.py3-none-win_amd64.whl
> python.exe -m pip install triton-2.0.0-cp310-cp310-win_amd64.whl
OK了,那边的训练PowerShell窗口里就不报错了。