MOSS icon indicating copy to clipboard operation
MOSS copied to clipboard

运行量化版本安装triton失败

Open SitaraJin opened this issue 2 years ago • 6 comments

在安装triton时老是提示下面两个错误: ERROR: Could not find a version that satisfies the requirement triton (from versions: none) ERROR: No matching distribution found for triton 查了下,网上回答是triton包不支持windows...这个能不能注释一下...或者有没有哪位大神有编译好的whl包?目前只找到适用于python3.10的,这里又要求python3.8

SitaraJin avatar Apr 24 '23 08:04 SitaraJin

from .custom_autotune import * 是这行代码的问题。不清楚这个哪里来的,希望作者能看一下

priest2018 avatar Apr 24 '23 08:04 priest2018

我试了 ubuntu18.4 triton2.1 也编译不过去,官方能否给个 量化版本的 docker 镜像啊,这个环境搭建废了好几天,也么成功。

937739823 avatar Apr 24 '23 09:04 937739823

我试了 ubuntu18.4 triton2.1 也编译不过去,官方能否给个 量化版本的 docker 镜像啊,这个环境搭建废了好几天,也么成功。

确实,我换成ubuntu,到调用triton一直不行,装上了triton也显示autotune不可用,一直提示安装triton

SitaraJin avatar Apr 24 '23 09:04 SitaraJin

我试了 ubuntu18.4 triton2.1 也编译不过去,官方能否给个 量化版本的 docker 镜像啊,这个环境搭建废了好几天,也么成功。

确实,我换成ubuntu,到调用triton一直不行,装上了triton也显示autotune不可用,一直提示安装triton

我这边是把models目录里的custom_autotune.py文件放在~/.cache/huggingface/modules/transformers_modules/local/ 目录下解决

cnsky2016 avatar Apr 24 '23 10:04 cnsky2016

triton装了也一直显示triton not installed. Run pip install triton to load quantized version of MOSS. NameError: name 'autotune' is not defined.

UTimeStrange avatar Apr 24 '23 11:04 UTimeStrange

经过测试,发现多次重新卸载pip uninstall triton后重新安装triton即可运行。

ChangxingSu avatar Apr 24 '23 14:04 ChangxingSu

经过测试,发现多次重新卸载pip uninstall triton后重新安装triton即可运行。

指定安装triton2.0.0后成功,默认安装是带post1后缀版本的,替换之后就可以了。

SitaraJin avatar Apr 25 '23 09:04 SitaraJin