Windows上安装Triton

本文提供了Windows系统下安装Mamba2及相关依赖（triton3.1.0、causal-conv1d1.4.0、mamba-ssm2.2.2）的详细教程。主要内容包括：安装NVIDIA驱动和CUDA12.4、配置VisualStudioBuildTools2022编译环境、设置系统环境变量、创建Conda环境并安装PyTorch，以及分步编译安装各组件。特别针对Windows平台常见的MSVC编译器堆空间不足问题，给出了通过读取显卡型号仅编译当前显卡架构的解决方案

【代码】Triton/window安装: triton-2.0.0-cp310-cp310-win_amd64.whl文件。

vcredist 是必需的（也称为“Visual C++ Redistributable for Visual Studio 2015-2022”，有大神强行在Windows平台上编译了 triton 的whl，参考博客【等功能均无法像Linux下正常运行，上述安装包只是从形式上完成编译。】，在下载路径下 pip install 安装 whl 文件即可。注意只需要下载单个组件，无需安装Visual Studio。但是，实测上述安装包里面 triton 核心的。修改环境变量，请参考博客：【

Triton 是一个开源项目，旨在为 AI 模型提供高效的编译优化。该项目支持在 Windows 操作系统上运行，使得在非自由平台上也能享受开源软件的便利。Triton 主要使用 C++ 进行开发，并且与 Python 深度集成，以提供易用的接口。 ## 2. 项目使用的关键技术和框架 - **C++**: 作为项目的主要编程语言，C++ 提供了高性能的执行环境。 - **Python**: ...

工作记录在 Windows 系统中安装 Triton 时，直接使用可能会遇到问题，因为 Triton 主要支持 Linux 系统。

Although technically Triton can be used alone, in the following let's assume you use it with PyTorch. Check your PyTorch version:尽管从技术上讲，Triton 可以单独使用，但在接下来的内容中，我们假设您将其与 PyTorch 一起使用。不支持 PyTorch 2.3 及更早版本。如果您使用的是低于 2.6 版本的 PyTorch，我建议您升级到 2.6 版本，因为该版本对。

本文摘要：针对Triton在Windows系统安装过程中出现的Python库查找失败、编译错误等问题，提供了详细的解决方案。关键步骤包括：1）参照GitHub库woct0rdho/triton-windows的指南；2）正确安装C++编译环境并设置环境变量路径；3）处理编译所需的Libs和include文件，将解压后的文件包合并到当前Python环境的Include文件夹中。特别提醒需注意ComfyUI与嵌入式Python的特殊配置要求。

这是因为 pypi 中的 triton 中没有适用于 Python 的版本。如果继续抛出如下异常，可能是版本不匹配，暂未找到合适的解决方法。通常是用来显卡加速的，可以尝试暂时卸载。

在Windows系统中，通过.whl文件安装Triton是一个相对直接的过程。

这里的原因在于，官方没有提供windows版本的triton，但是有一个社区版的triton-windows.uv是一个python项目管理工具，如果你没有用uv配置环境的化，可以去掉uv，直接使用下面的。使用ComfyUI时，遇到了缺少triton包的问题，当我试图安装时，显示没有wheels。解决方法是安装这个社区版的triton-windows.

不要再去看其他啥教程了，因为有大佬在维护window平台的triton。