ROCm 开发人员中心
在这里可以找到各种 ROCm 开发人员资源,涵盖文档、培训课程、网络研讨会、最新博客等。
AMD ROCm 软件是什么:AMD GPU 编程
ROCm 是一个开放式的生产级软件平台,其中包含驱动程序、库和开发者工具,旨在助力 AMD GPU 编程。
ROCm 可赋能 HPC 和 AI 工作负载,支持开发者使用 PyTorch 等常用框架进行大规模模型训练和推理。
新增功能
开始使用 ROCm 进行构建
根据工作负载选择合适的 GPU
AMD Instinct GPU 专为训练 AI 模型、运行大规模推理服务和 HPC 仿真而打造。
AMD Radeon GPU 专为 AI 驱动的游戏、实时图形处理、创意工作流程和本地 LLM 而打造。
轻松从 CUDA 迁移至 ROCm,几乎无需更改代码
您可以直接在 AMD GPU 上运行多种工作负载,而无需更改代码。当需要更改代码时,HIPIFY 工具可帮助您将现有 CUDA 代码转换为兼容 ROCm 的 HIP 代码,确保充分减少中断。
加入 AI 开发人员计划
获取 AMD Developer Cloud 免费额度、专属培训课程、每月 AMD GPU 抽奖活动以及社区支持,旨在助您顺利开展 AI 开发工作。
常见问题解答
ROCm 和 CUDA 都是 GPU 编程平台,但其设计理念截然不同。
ROCm 是一个完全开源的生产级软件栈,兼容多种 AMD GPU。ROCm 支持多种强大工具、AI 模型和框架,包括 PyTorch、TensorFlow、JAX 等。该平台具备高度灵活性,可针对特定环境与需求进行自定义、修改和优化,同时可依托充满活力的开放生态系统加速推动创新。
CUDA 是 NVIDIA 的专有平台,与 NVIDIA 硬件紧密集成。虽然 CUDA 非常成熟,但会将开发者锁定在其封闭的生态系统中,从而限制自由选择权。
多款 AMD Instinct GPU、AMD Radeon GPU 以及部分 AMD 锐龙 AI 处理器中的集成 GPU (iGPU) 均兼容 ROCm 软件,为开发者提供了广泛的选择。请查看以下兼容性矩阵,详细了解计算工作负载和图形工作负载对应的兼容 GPU。您还可以找到支持的开源框架、库、工具及相应的系统要求。
ROCm 主要用于 GPU 加速的 AI 和 HPC 工作负载,特别是运行 PyTorch、TensorFlow 和 JAX 等深度学习框架以进行训练、微调和推理。
主要应用场景包括:
- 跨多个 GPU 的分布式训练
- 通过 vLLM 和 SGLang 等经过优化的引擎提供推理服务
- 依托 Hugging Face 和其他开源生态系统进行模型优化
开发者只需进行少量更改,即可在 AMD GPU 上运行现有代码库。
通过 AMD 开发者社区论坛和 AMD 开发者 Discord 服务器,您可以直接与其他开发者和 AMD 专家沟通交流,共同讨论问题并帮助解决问题。您还可以在 GitHub 上提交问题。
订阅接收 ROCm 资讯
订阅接收 ROCm 新闻资讯,及时掌握最新版本发布动态及相关实用技巧。