Repolode
提交项目
KomputeProject

KomputeProject/kompute

基于 Vulkan 的通用 GPU 计算框架，提供 C++ 与 Python 接口，可在 AMD、高通、NVIDIA 等多厂商显卡上运行。

可替代:CUDA

在 GitHub 查看官网
Star
2,581
Fork
202
未关闭 Issue
79
最近提交
3个月前

这是什么？

Kompute 是一个通用 GPU 计算框架，以 Vulkan 作为后端，同一份代码可以在 AMD、高通、NVIDIA 等不同厂商的显卡上运行。它把 Vulkan 计算接口的大量样板代码封装为 Manager、Tensor、Sequence 和 Algorithm 等组件，开发者可以用更少的代码完成数据上传到 GPU、执行 shader 并读回结果的流程。

框架同时提供 C++ SDK 和 Python 包：C++ 接口便于底层优化和扩展，Python 包则适合交互式实验。操作可以录制到 Sequence 中异步提交，并能分配到多个 GPU 队列族上并行执行。它设计为与已有 Vulkan 应用共存，而不是隐藏 Vulkan 接口。

在移动端，Kompute 支持在 Android 上动态加载 Vulkan 库，并提供 C++ 头文件的 NDK 封装。项目中包含逻辑回归、Android 应用和 Godot 游戏引擎集成等示例。

核心功能

  • 01通过 Vulkan 支持 AMD、高通、NVIDIA 等多厂商显卡。
  • 02提供用于底层优化的 C++ SDK 和便于交互使用的 Python 包。
  • 03支持异步执行，并可在多个 GPU 队列族上并行运行操作。
  • 04通过 Tensor 显式管理 GPU 与主机内存。
  • 05提供 Android NDK 封装和 Vulkan 动态加载，便于移动端构建。
  • 06可与已有 Vulkan 代码共存，不强制替换底层接口。

适用场景

  • 在没有 CUDA 的 AMD 或高通等硬件上运行机器学习或数值计算任务。
  • 为已有的 Vulkan 渲染器或游戏引擎增加 GPU 计算能力。
  • 在 Android 应用中实现端侧 GPU 推理或数据处理。
  • 先用 Python 原型验证 GPU 算法，再迁移到 C++ 实现。

使用前须知

  • 需要支持 Vulkan 的 GPU 和驱动程序，因为它基于 Vulkan 构建。
  • README 中的 Python 徽章仅标注支持 3.7 至 3.9 版本。

收录于专题

Android 应用与手机工具

标签

#vulkan#gpu-computing#gpgpu#machine-learning#cross-platform#c-plus-plus#python#mobile#cpp#deep-learning

常见问题

kompute 是什么？
基于 Vulkan 的通用 GPU 计算框架，提供 C++ 与 Python 接口，可在 AMD、高通、NVIDIA 等多厂商显卡上运行。 Kompute 是一个通用 GPU 计算框架，以 Vulkan 作为后端，同一份代码可以在 AMD、高通、NVIDIA 等不同厂商的显卡上运行。它把 Vulkan 计算接口的大量样板代码封装为 Manager、Tensor、Sequence 和 Algorithm 等组件，开发者可以用更少的代码完成数据上传到 GPU、执行 shader 并读回结果的流程。
kompute 可以免费使用吗？用的什么许可证？
kompute 以 Apache-2.0 许可证开源。用于商业产品前，请先确认许可证条款。
kompute 还在维护吗？
半年内有更新。kompute 最近一次提交是在 2026年8月15日。
kompute 适合用来做什么？
在没有 CUDA 的 AMD 或高通等硬件上运行机器学习或数值计算任务。 为已有的 Vulkan 渲染器或游戏引擎增加 GPU 计算能力。 在 Android 应用中实现端侧 GPU 推理或数据处理。 先用 Python 原型验证 GPU 算法，再迁移到 C++ 实现。
在哪里获取 kompute？
源代码和安装说明在 GitHub 上：https://github.com/KomputeProject/kompute

概要由 AI 根据项目 README 生成；GitHub 数据每日更新。

相关项目