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 解决。



声明

本文内容仅代表作者观点,或转载于其他网站,本站不以此文作为商业用途
如有涉及侵权,请联系本站进行删除
转载本站原创文章,请注明来源及作者。