Python｜Windows 系统安装 triton 的方法

Python资料网推荐Python｜Windows 系统安装 triton 的方法这篇文章给大家，欢迎收藏Python资料网享受知识的乐趣

问题现象

若未安装，则在运行调用了该仓库的 Python 脚本时，会报错如下：

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 文件即可：

pip install triton-2.1.0-cp311-cp311-win_amd64.whl

后续问题

如果继续抛出如下异常，可能是版本不匹配，暂未找到合适的解决方法。

ModuleNotFoundError: No module named 'triton.common'

如果在 unsloth 用到，那么通常因为 unsloth 通常是用来显卡加速的，可以尝试暂时卸载 unsloth 解决。

版权声明：本文来自互联网用户投稿，该文观点仅代表作者本人，不代表本站立场。本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。如若内容造成侵权/违法违规/事实不符，请联系邮箱：jacktools123@163.com进行投诉反馈，一经查实，立即删除！