最初写于2024-6-4,文后共两次更新,最后更新于2025-03-19
以前依据此博主文章
分析解决【No module named ‘triton‘】的问题_error caught was: no module named 'triton-CSDN博客
解决No module named 'triton'的问题是可以成功的
据安装过的网友反馈2.0.0这个编译版本可能有问题,会造成训练异常。
但随着xformers的版本更新,最近在CUDA12.1支持的xformers0.0.23版本下,安装此2.0.0版本的triton无法使用,安装后显示
WARNING:xformers:A matching Triton is not available, some optimizations will not be enabled.
Error caught was: DLL load failed while importing libtriton: 找不到指定的模块。
xformers version: 0.0.23.post1
于是顺手查了下,找到最新编译2.1.0版本
Python 3.10 https://huggingface.co/madbuda/triton-windows-builds/resolve/main/triton-2.1.0-cp310-cp310-win_amd64.whl
https://huggingface.co/madbuda/triton-windows-builds/resolve/main/triton-2.1.0-cp311-cp311-win_amd64.whl |
我是用3.10版本,下到本地运行如下命令
pip install triton-2.1.0-cp310-cp310-win_amd64.whl
pip install triton-2.1.0-cp310-cp310-win_amd64.whl
重新开启后即不再显示错误
对比了下2.0.0的编译版本才10来M,而2.1.0的版本有300M
可能也印证了2.0.0版本确实可能存在问题
2024-12-21更新,分享几个已编译文件【链接在最后】
部分网友反应给出的链接在外网,下载有些困难。
这边分享几个已经下载好的2.1.0的编译版本,大小介于300M-800M之间
有对应python3.10和3.11两个环境,每个环境都有两个版本,测试都正常
建议下文件最大的版本,不能用再用小点的版本
2025-03-19更新,更新最新3.0.0-3.2.0版本 【py310-312】
torch2.4.0建议使用3.0.0-3.1.0版本,torch2.6.0建议使用3.2.0版本
当然如果出问题可自行尝试替换版本,注意对应python版本
夸克盘分享地址:夸克网盘分享