NVIDIA/aerial-cuda-accelerated-ran
An SDK (Software Development Kit) for building commercial-grade, AI-native, 3GPP, and O-RAN compliant 5G/6G gNB software on NVIDIA-accelerated computing platforms.
NVIDIA Aerial™ CUDA 加速 RAN
是什么 – 一个开源 SDK,将 5G 无线接入网(RAN)的繁重计算任务迁移至 NVIDIA GPU 上。它提供了物理层(L1)和 MAC 层(L2)的 CUDA 实现,并附带 Python 绑定,使研究人员能够轻松将该栈集成到 AI/ML 框架中。
为何重要 – 5G 及未来的 6G 系统需要大量计算能力来处理 LDPC/Polar 信道编码、大规模 MIMO 波束成形和实时调度等任务。通过将这些任务卸载到 GPU,该 SDK 实现了:
- 快速仿真真实的 5G 波形。
- 快速原型化 AI 增强的调度或信道估计算法。
- 可扩展的容器化部署,适用于研究集群或边缘服务器。
核心组件
| 组件 | 作用 | 亮点 |
|---|---|---|
| cuPHY | CUDA 加速物理层(L1) | LDPC 与 Polar 编码、调制/解调、MIMO 处理、信道估计 —— 全部在 GPU 上运行。 |
| cuMAC | CUDA 加速 MAC 层(L2) | 高吞吐量调度器,用于资源分配和定时。 |
| pyAerial | Python API | 向 Python 暴露 cuPHY/cuMAC,便于从 TensorFlow、PyTorch 或 Sionna 通信 ML 库中调用。 |
| 5GModel | 参考波形生成器 | 基于 MATLAB 的脚本,生成测试向量并验证 GPU 内核是否符合 3GPP 规范。 |
| 容器环境 | NVIDIA NGC 上的 Docker 镜像 | 包含所有依赖项的预构建容器,开箱即用。 |
仓库结构(概览)
cuPHY/– 物理层的 GPU 内核。cuMAC/– MAC 调度器的 GPU 内核。pyaerial/– Python 绑定和辅助 ML 工具。5GModel/– 测试向量生成和验证工具。testBenches/– 性能测量和正确性测试脚本。cuPHY-CP/– 控制面连接(前传驱动、RU 模拟器等)和容器构建脚本。
快速入门
# 克隆并包含子模块
git clone --recurse-submodules https://github.com/NVIDIA/aerial-cuda-accelerated-ran.git
cd aerial-cuda-accelerated-ran
# 安装 Git-LFS 并拉取大文件
sudo apt install git-lfs && git lfs pull
# 启动开发容器(包含 CUDA、驱动和所有库)
./cuPHY-CP/container/run_aerial.sh
# 在容器内构建 SDK
./testBenches/phase4_test_scripts/build_aerial_sdk.sh
容器镜像也可在 NVIDIA NGC 上获取,适用于 CI/CD 流水线。
典型应用场景
- 研究 – 通过
pyAerial插入自定义 Python 代码,评估新的 AI 驱动的信道估计或调度算法,同时保留高性能 GPU 内核。 - 仿真 – 生成具有真实 PHY 行为的大规模 5G NR 业务流量记录,用于网络级研究。
- 原型部署 – 在 GPU 支持的服务器上运行完整 RAN 栈,用于边缘计算演示或早期 6G 测试床。
支持与贡献
- 该项目由 NVIDIA 维护。问题可在 GitHub 上提交;安全报告请通过
SECURITY.md中描述的私密披露流程提交。 - 目前 不接受贡献(外部开发者仅可读取仓库)。
许可证 – Apache License 2.0(见 LICENSE)。部分第三方组件有各自许可证(列于 ATTRIBUTION.rst)。
引用 – 若您发表使用此 SDK 的研究成果,请引用提供的 BibTeX 条目。
以上所有信息均直接来自仓库的 README;未推断任何额外功能。
相关
- 项目
- 项目
- 项目
- 项目
- 项目