分析解决【No module named ‘triton‘】的问题

(一)现象

在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窗口里就不报错了。


PS:继续吐槽一下咱的网络,好像56kbps猫上网的时代:
分析解决【No module named ‘triton‘】的问题_第1张图片

你可能感兴趣的:(Python,人工智能/机器学习,Windows,triton,LoRA,python,windows,训练)