OriginDL:用C++/CUDA手写一个深度学习框架实现训练与推理

#推荐一个值得做的AI项目#推荐我的个人github项目:https://:github.com/jinbooooom/OriginDL。OriginDL 是一个完全从零开始构建的深度学习框架,采用 C++ 实现。OriginDL 不仅实现了高层的自动微分系统,更从最底层的矩阵运算开始,使用 CUDA 手写了 GPU 加速的矩阵计算核心,并在此基础上构建了完整的自动求导引擎和神经网络模块。目前已在框架上实现了线性回归与 MNIST 手写数字识别的训练示例,以及基于 PNNX 的 YOLOv5 目标检测推理,用于验证自动求导与推理链路。

OriginDL 提供了类似 PyTorch 的 API 接口,并配有详细的设计文档和用户指南,帮助对框架底层实现感兴趣的朋友深入理解深度学习框架的底层实现原理。更多细节可以在项目首页的README中看到。https://:github.com/jinbooooom/OriginDL
全部评论

相关推荐

评论
点赞
收藏
分享

创作者周榜

更多
牛客网
牛客网在线编程
牛客网题解
牛客企业服务