AMD ROCm 是 AMD 的开源 GPU 计算平台,定位为面向 AI、HPC 和领域特定工作负载的端到端生态系统,由编译器、运行时和库组成。
用途 ROCm 用于在 AMD 硬件上进行 GPU 计算,覆盖从基础数学与计算库、通信原语、HIP 运行时,到性能分析与调试工具等基础能力,并支持 AI 与 HPC 工作负载。该平台开源、跨平台(Linux 和 Windows),针对 AMD Instinct™、AMD Radeon™ 和 AMD Ryzen™ AI 设备优化。
主要能力 根据官方页面,ROCm 提供以下文档与资源板块:
- 核心 SDK:面向 AMD 硬件 GPU 计算的基础库、运行时和工具,包括数学与计算库、通信原语、HIP 运行时、性能分析与调试工具等,并提供 ROCm Core SDK 安装与 ROCm 10.1.0 兼容性信息。
- AI 生态:提供全栈文档与配方,用于在 AMD GPU 上使用主流 ROCm 支持的框架部署 AI 工作负载,涵盖深度学习框架 PyTorch、JAX,以及推理方案 vLLM、SGLang。
- GPU 系统:面向大规模 AMD Instinct GPU 的部署与运维指南,包括 AMD GPU 驱动(amdgpu)安装、集群管理、GPU 分区、监控、虚拟化、云部署和容器,以及 Kubernetes 网络算子(Network Operator)。
- 领域加速库:开源集合,用于构建高性能领域特定应用,目前列出 ROCm Data Science、ROCm Life Science 和 ROCm Simulation。
使用方式 官方页面提供安装 ROCm Core SDK 的文档入口、ROCm 10.1.0 兼容性信息、Core SDK 文档入口、AI 生态文档入口和 GPU 系统文档入口。具体安装命令、配置步骤和代码示例需从对应文档中获取。
适用场景 适用于在 AMD Instinct、AMD Radeon 和 AMD Ryzen AI 设备上进行 GPU 计算的开发者与运维人员,包括 AI 模型训练与推理、HPC 计算、领域特定高性能应用(数据科学、生命科学、仿真),以及大规模 GPU 集群的部署与运维。
限制 当前官方文本未提供支持的完整硬件列表、具体版本兼容矩阵、许可条款细节、价格或收费信息、性能数据,也未提供具体安装命令或代码示例。