网易首页 > 网易号 > 正文 申请入驻

8GB显存起步,一名本科生用CUDA手搓LLM推理引擎

0
分享至

一名计算机科学专业的本科生,最近开始了一个叫 Project EREN 的项目:用 C++ 和 CUDA 从零搭建一个 LLM 推理引擎。他把第一阶段的目标定得很具体——先在自己的 NVIDIA RTX 4060 Laptop GPU(8 GB 显存)上跑通,从 GPT-2 这类小型 decoder-only transformer 起步,先把正确性立住,再逐步走向更强的模型和更高效的推理。


这不是"调个框架加载模型"的路线。他明确说,自己想弄明白的是那些被 PyTorch 和成熟推理运行时封装起来的底层:transformer 的运算怎么映射到 GPU kernel 上,模型权重和中间激活在显存里经历了什么,自回归逐 token 生成在内部是怎么运转的,以及为什么在推理过程中内存带宽和 kernel 启动开销会变得重要。他的做法是自己实现、测试、跑基准、再优化,而不是只停留在阅读概念。

长期目标也写得很清楚:支持多个并发推理请求、追求高吞吐,同时把底层的 GPU kernel、内存管理、调度和执行策略吃透。再往后,他希望 EREN 能成为某个基于 RAG 的 AI agent 的推理后端。他也把话说在前面——眼下并不打算和成熟的推理引擎正面竞争,重点是通过亲手造一个来理解这类系统怎么工作。

技术栈与当前进度

开发环境是 C++17 加 CUDA C++,跑在 WSL2 上的 Ubuntu 里,构建用 CMake,测试用 CTest。后续还会引入 Python、PyTorch 和 Hugging Face Transformers 作为参考实现,用来对照验证正确性。核心推理运行时则坚持用 C++ 和 CUDA 写。

目前完成的是 Phase 0,也就是把 C++/CUDA 的开发地基打牢。已经落地的部分包括:

  • 项目结构与构建系统:仓库按 kernels、源码、头文件、测试、基准和文档分目录组织,CMake 负责构建 C++ 与 CUDA 目标。
  • CUDA 向量加法:用一个基础 kernel 验证显存分配、主机到设备传输、kernel 启动和结果校验这条链路。
  • GPU reduction:实现了基于共享内存的多阶段归约基线,用来理解并行归约和跨线程块协作。
  • CUDA 错误处理:封装了一个可复用的错误检查辅助工具,方便定位和调试运行时故障。
  • 测试与基准基础:用 CTest 加了冒烟测试,并建立了最初的 reduction 基准目标。

他给出的结果是:一次干净的构建成功,配置的四个测试全部通过。不过他也坦承,推理引擎本身还没跑起来,这一阶段只是为下一步铺路。

分阶段推进,不直接跳进优化

他选择按阶段推进,而不是一上来就写优化过的 kernel。Phase 1 是先做 Python/PyTorch 参考实现:从 GPT-2 入手,研究它的前向传播,检查权重和张量形状,并生成用于正确性测试的参考输出。Phase 2 是第一个 CUDA 推理实现:写出最初的 CUDA 算子,并把它们串成一次完整的 GPT-2 前向传播。更后面的阶段才涉及优化与服务化,包括优化矩阵乘法、量化、GPU profiling、内存管理、批处理、并发请求和推理调度。整条路径的意图很明确——从"能正确跑"走向"可度量、够高效"的运行时。

他公开记录这个项目,也不只是为了发进度公告。后续会分享自己在实现什么、为什么这么做,CUDA 概念和工程决策,遇到的 bug、失败的实验和踩过的坑,以及基准测试和性能对比。他还特意提到,会尽量区分哪些是已经跑通的、哪些是实际测过的、哪些还在实验当中。

EREN 仍处在早期,要学的东西还有很多。下一步,他会先在 PyTorch 里搭出 GPT-2 的参考实现,然后再进入 CUDA 推理。

特别声明:以上内容(如有图片或视频亦包括在内)为自媒体平台“网易号”用户上传并发布,本平台仅提供信息存储服务。

Notice: The content above (including the pictures and videos if any) is uploaded and posted by a user of NetEase Hao, which is a social media platform and only provides information storage services.

相关推荐
热点推荐
红果短剧,正在悄悄透支整整一代人的未来。

红果短剧,正在悄悄透支整整一代人的未来。

老陆不老
2026-10-22 05:10:02
今年国产高端已经完了,苹果彻底卖爆了

今年国产高端已经完了,苹果彻底卖爆了

六子吃凉粉
2026-10-23 12:20:41
周恩来在劳山遇刺,11名警卫全部牺牲,调查后才知凶手竟大有来历

周恩来在劳山遇刺,11名警卫全部牺牲,调查后才知凶手竟大有来历

小哥很OK
2025-11-17 21:02:50
绝境绝不低头!乌克兰女机枪手血战三天三夜,成首位女英雄

绝境绝不低头!乌克兰女机枪手血战三天三夜,成首位女英雄

老马拉车莫少装
2026-10-22 00:00:26
中国癌症最新数据出炉:40岁是关键分水岭,60岁是癌症负担重灾区,肺癌新发、死亡双双登顶

中国癌症最新数据出炉:40岁是关键分水岭,60岁是癌症负担重灾区,肺癌新发、死亡双双登顶

医护健康科普
2026-10-24 09:06:05
一波未平一波又起!郭德纲再次被举报,官方已回应,又是即兴改编

一波未平一波又起!郭德纲再次被举报,官方已回应,又是即兴改编

天马幸福的人生
2026-10-22 15:28:37
39岁郑欣宜在沈殿霞墓前哭得像个孩子,6000万信托,母亲算尽一切

39岁郑欣宜在沈殿霞墓前哭得像个孩子,6000万信托,母亲算尽一切

江启
2026-10-23 09:20:49
我不是软蛋!申京29+7+7硬刚文班!杜兰特死神,阿门逆天绝杀大帽

我不是软蛋!申京29+7+7硬刚文班!杜兰特死神,阿门逆天绝杀大帽

Tracy的篮球博物馆
2026-10-24 12:29:08
黄金又跌破4000美元了

黄金又跌破4000美元了

每日经济新闻
2026-10-24 09:45:14
美国人想不通:一艘服役仅仅6个月055万吨大驱,就敢抵近日本近海

美国人想不通:一艘服役仅仅6个月055万吨大驱,就敢抵近日本近海

像梦一场a
2026-10-23 23:39:03
赔率排第3!曝皇马坚信姆巴佩将获金球奖 79岁佛爷率豪华阵容出席

赔率排第3!曝皇马坚信姆巴佩将获金球奖 79岁佛爷率豪华阵容出席

风过乡
2026-10-24 07:13:43
阴间真的存在吗?科学家给出了答案,看完你就明白了

阴间真的存在吗?科学家给出了答案,看完你就明白了

心中的麦田
2026-10-22 21:54:56
这下,轮到躺平收租的广东房东开始慌了!

这下,轮到躺平收租的广东房东开始慌了!

童童聊娱乐啊
2026-10-24 04:00:54
普京:克里米亚,乌东四州本就是俄罗斯的,乌克兰根本没损失领土

普京:克里米亚,乌东四州本就是俄罗斯的,乌克兰根本没损失领土

南生今世说
2026-10-23 09:19:16
秘而不宣!苏提达儿子长期在瑞士生活,深耕泰语和王室礼仪,官方态度耐人寻味

秘而不宣!苏提达儿子长期在瑞士生活,深耕泰语和王室礼仪,官方态度耐人寻味

秋风妃
2026-10-23 19:51:56
利空!300409,百亿新材料“小巨人”,Q3净利下滑超94%

利空!300409,百亿新材料“小巨人”,Q3净利下滑超94%

上市之家
2026-10-24 03:07:16
火箭客场大战马刺,球员评分:4人满分、4人达标、1人崩盘

火箭客场大战马刺,球员评分:4人满分、4人达标、1人崩盘

十三哥侃大山
2026-10-24 11:13:43
一个贪财一个好色!三婚娶小37岁徒弟,81生女84生儿,现活成这样

一个贪财一个好色!三婚娶小37岁徒弟,81生女84生儿,现活成这样

兵卒史
2026-10-13 00:43:48
骑士爆发矛盾,米切尔公开质疑哈登,希望他能认真防守

骑士爆发矛盾,米切尔公开质疑哈登,希望他能认真防守

篮球小方
2026-10-24 12:09:25
一女孩在F4演唱会因外形酷似大S引热议,本人回应:“长相是自然的,没有去do脸,也没有化仿妆”

一女孩在F4演唱会因外形酷似大S引热议,本人回应:“长相是自然的,没有去do脸,也没有化仿妆”

韩小娱
2026-10-22 07:02:07
2026-10-24 13:20:49
字节漫游指南
字节漫游指南
有态度网友ytd
211文章数 128关注度
往期回顾 全部

科技要闻

写程序、做3D,新Kimi还有多少本事?

头条要闻

媒体:法国爆发暴力抗议活动 中国或被指认为主要原因

头条要闻

媒体:法国爆发暴力抗议活动 中国或被指认为主要原因

体育要闻

从乐透秀到总冠军再到没工作 一共要几年?

娱乐要闻

一家人就林依轮一个中国人 其余全外国

财经要闻

叫"保障"不叫"保险":租车平台的文字游戏

汽车要闻

设计换新/智舱升级 2027款博越限时价7.89万起

态度原创

教育
艺术
旅游
手机
公开课

教育要闻

初中太努力不是好事!家长拍下高中儿子休学,过来人劝她看清现实

艺术要闻

16位中国当代画家油画作品

旅游要闻

新东方入局铁路旅游,“新东方·霞客号”列车12月自北京站始发

手机要闻

苹果首款折叠iPhone Duo开放到店自提预约服务

公开课

李玫瑾:为什么性格比能力更重要?

无障碍浏览 进入关怀版