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
相关推荐
colus_SEU20 分钟前
【循环神经网络6】LSTM实战——基于LSTM的IMDb电影评论情感分析
人工智能·rnn·深度学习·神经网络·lstm
无风听海37 分钟前
神经网络之损失函数
深度学习·神经网络·机器学习
JJJJ_iii1 小时前
【深度学习01】快速上手 PyTorch:环境 + IDE+Dataset
pytorch·笔记·python·深度学习·学习·jupyter
2401_841495643 小时前
【计算机视觉】分水岭实现医学诊断
图像处理·人工智能·python·算法·计算机视觉·分水岭算法·医学ct图像分割
diqiudq5 小时前
用AMD显卡节省nVidia显卡显存占用
linux·深度学习·ubuntu·显存释放
格林威6 小时前
常规可见光相机在工业视觉检测中的应用
图像处理·人工智能·数码相机·计算机视觉·视觉检测
MUTA️6 小时前
论文速览:从ConvNeXt 到 ConvNeXt V2
人工智能·深度学习
格林威8 小时前
短波红外相机在工业视觉检测中的应用
人工智能·深度学习·数码相机·算法·计算机视觉·视觉检测
EEG小佬8 小时前
KAN(Kolmogorov-Arnold Networks)通俗理解
人工智能·python·深度学习·神经网络
扶尔魔ocy8 小时前
【QT常用技术讲解】multimedia实现指定分辨率打开摄像头
图像处理·qt