ModuleNotFoundError: No module named ‘triton‘

最近使用Kolmogorov-Arnold Transformation测试程序,代码中有一段引用

python 复制代码
from kat_rational import KAT_Group

其中需要

python 复制代码
import triton

由于我用的windows系统进行测试,出现了如下问题

python 复制代码
Traceback (most recent call last):
  File "C:/Users/64883/Desktop/Cy/DoubleKATBlock.py", line 4, in <module>
    from kat_rational import KAT_Group
  File "C:\Users\64883\Desktop\Cy\kat_rational\__init__.py", line 2, in <module>
    from .kat_1dgroup_triton import RationalTriton1DGroup, KAT_Group
  File "C:\Users\64883\Desktop\Cy\kat_rational\kat_1dgroup_triton.py", line 4, in <module>
    from .rational_triton import RationalTriton1DGroup
  File "C:\Users\64883\Desktop\Cy\kat_rational\rational_triton.py", line 2, in <module>
    import triton
ModuleNotFoundError: No module named 'triton'

凭着直觉,直接输入了

python 复制代码
pip install triton

但是没有成功,出现了如下问题

python 复制代码
Looking in indexes: https://pypi.tuna.tsinghua.edu.cn/simple
ERROR: Could not find a version that satisfies the requirement triton (from versions: none)
ERROR: No matching distribution found for triton

查了一下资料,发现triton 是 Meta(Facebook)开发的用于高性能深度学习内核编写的库,而目前只有在支持的环境(通常是 Linux + NVIDIA GPU)上才能正确编译和运行。因此在 Windows 上安装和使用 triton 可能会失败或无法使用。

因此,我们只需要将代码上传到Linux系统即可。

额外问题

python 复制代码
ImportError: cannot import name 'cached_property' from 'functools' (/home/sys120-1/miniconda3/envs/pyt_3.7/lib/python3.7/functools.py)

解决方法:Python 版本是 3.7,而 functools.cached_property 是从 Python 3.8 才引入的标准库功能。在 Python 3.7 中该属性不存在,导致 triton 加载失败。

python 复制代码
AttributeError: module 'triton.language' has no attribute 'fma'

解决方法:说明当前安装的 triton 版本过旧,不支持 tl.fma,升级 Triton 到最新版

python 复制代码
pip install --upgrade triton
相关推荐
小白狮ww2 小时前
dots.ocr 基于 1.7B 参数实现多语言文档处理,性能达 SOTA
人工智能·深度学习·机器学习·自然语言处理·ocr·小红书·文档处理
无风听海12 小时前
神经网络之窗口大小对词语义向量的影响
人工智能·深度学习·神经网络
sali-tec12 小时前
C# 基于halcon的视觉工作流-章52-生成标定板
开发语言·图像处理·人工智能·算法·计算机视觉
Tiandaren12 小时前
自用提示词02 || Prompt Engineering || RAG数据切分 || 作用:通过LLM将文档切分成chunks
数据库·pytorch·深度学习·oracle·prompt·rag
阿水实证通13 小时前
面向社科研究者:用深度学习做因果推断(二)
深度学习·1024程序员节·因果推断·实证分析·科研创新
文火冰糖的硅基工坊14 小时前
[人工智能-大模型-83]:模型层技术 - 前向预测:神经网络是如何产生涌现智能的?背后的本质是什么?
人工智能·深度学习·神经网络
子不语18014 小时前
深度学习——IDE之Jupyter
人工智能·深度学习·jupyter
高洁0115 小时前
【无标题】大模型-模型压缩:量化、剪枝、蒸馏、二值化 (2
人工智能·python·深度学习·神经网络·知识图谱
WWZZ202515 小时前
快速上手大模型:深度学习2(实践:深度学习基础、线性回归)
人工智能·深度学习·算法·计算机视觉·机器人·大模型·slam
学技术的大胜嗷16 小时前
图像归一化:OpenCV 高效映射 [0,255] → [-1,1] 性能实测
图像处理·opencv·计算机视觉