在win10 下安装 triton
pip install triton
安装过程因为没有找到这个库报错,如下图:
triton 本般是开源的,打开开源项目地址:GitHub - openai/triton: Development repository for the Triton language and compiler
从项目主页可以看到只兼容linux平台
只能修改源码 在window平台修改做安装如下:
git clone https://github.com/openai/triton.git;
cd triton;
pip install ninja cmake wheel; # build-time dependencies
pip install -e python
由于时间问题,直接找了一个已经处理好支持win10的包使用。
https://huggingface.co/r4ziel/xformers_pre_built/blob/main/triton-2.0.0-cp310-cp310-win_amd64.whl
csdn下载方式:https://download.csdn.net/download/qyhua/88831509
下载后安装:
pip install F:\ai\triton-2.0.0-cp310-cp310-win_amd64.whl
安装成功如下图: