Python|Windows 系统安装 triton 的方法
长行 2024-07-23 17:05:01 阅读 68
问题现象
若未安装,则在运行调用了该仓库的 Python 脚本时,会报错如下:
<code>ModuleNotFoundError: No module named 'triton'
在 Windows 系统中,如果直接使用 pip 安装,会报错如下:
pip install triton
ERROR: Could not find a version that satisfies the requirement triton (from versions: none)
ERROR: No matching distribution found for triton
这是因为 pypi 中的 triton 中没有适用于 Python 的版本。
解决方法
Step 1|在 HuggingFace 上下载 triton 的 Windows 包:https://hf-mirror.com/madbuda/triton-windows-builds
Step 2|在下载路径下 pip install 安装 whl 文件即可:
<code>pip install triton-2.1.0-cp311-cp311-win_amd64.whl
后续问题
如果继续抛出如下异常,可能是版本不匹配,暂未找到合适的解决方法。
<code>ModuleNotFoundError: No module named 'triton.common'
如果在 unsloth
用到,那么通常因为 unsloth
通常是用来显卡加速的,可以尝试暂时卸载 unsloth
解决。
声明
本文内容仅代表作者观点,或转载于其他网站,本站不以此文作为商业用途
如有涉及侵权,请联系本站进行删除
转载本站原创文章,请注明来源及作者。