nemo-mbridge-perf-cuda-graphs · AI Skill 用途与安装 · iSkills.cc
在Megatron Bridge中配置CUDA图捕获以优化GPU训练性能。
分类：AI 与 Agent
来源仓库：nvidia/skills
适合用来做什么
- 请在 Megatron Bridge 中验证 CUDA 图捕获。
- 我需要使用 CUDA 图捕获 Transformer Engine 范围的注意力模块。
不适合的情况
- 不适合与“AI 与 Agent”无关的任务。
安装 nemo-mbridge-perf-cuda-graphs
npx skills add https://www.iskills.cc/skills/nvidia/skills/nemo-mbridge-perf-cuda-graphs/SKILL.md在本机终端执行安装命令。安装前请阅读原始说明，确认所需工具、账号和权限。