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
相关推荐
叶子爱分享3 小时前
计算机视觉与图像处理的关系
图像处理·人工智能·计算机视觉
张较瘦_3 小时前
[论文阅读] 人工智能 | 深度学习系统崩溃恢复新方案:DaiFu框架的原位修复技术
论文阅读·人工智能·深度学习
cver1233 小时前
野生动物检测数据集介绍-5,138张图片 野生动物保护监测 智能狩猎相机系统 生态研究与调查
人工智能·pytorch·深度学习·目标检测·计算机视觉·目标跟踪
学技术的大胜嗷3 小时前
离线迁移 Conda 环境到 Windows 服务器:用 conda-pack 摆脱硬路径限制
人工智能·深度学习·yolo·目标检测·机器学习
Akttt6 小时前
【T2I】R&B: REGION AND BOUNDARY AWARE ZERO-SHOT GROUNDED TEXT-TO-IMAGE GENERATION
人工智能·深度学习·计算机视觉·text2img
点我头像干啥8 小时前
用 PyTorch 构建液态神经网络(LNN):下一代动态深度学习模型
pytorch·深度学习·神经网络
小白狮ww8 小时前
VASP 教程:VASP 机器学习力场微调
人工智能·深度学习·机器学习
呆头鹅AI工作室9 小时前
[2025CVPR]SEEN-DA:基于语义熵引导的领域感知注意力机制
人工智能·深度学习·机器学习
加油吧zkf9 小时前
目标检测新纪元:DETR到Mamba实战解析
图像处理·人工智能·python·目标检测·分类
西柚小萌新10 小时前
【深度学习:进阶篇】--4.3.seq2seq与Attention机制
人工智能·深度学习