报错原因:2024.5.21之后, triton.ops 被移动到另一个工程 triton-lang/kernels中。

参考链接:官方解释
解决方案:换用2024.5.21之前发布的版本。

pip3 install triton==2.3.0

Logo

汇聚全球AI编程工具,助力开发者即刻编程。

更多推荐