ROCm是由AMD開發的開源GPU計算生態系統,主要套用於人工智慧、機器學習、科學計算及數據中心加速等領域。該系統支持HIP編程模型,提供Triton編譯器、HIPify代碼轉換工具等開發套件,兼容PyTorch、TensorFlow等主流AI框架。
2025年4月發布的ROCm 6.4.0版本新增對Radeon PRO W7800 48GB顯示卡的支持,同年6月推出的ROCm 7版本較前代實現推理性能提升最高3.8倍、訓練性能提升3倍,新增FP4/FP6低精度計算和分散式推理功能。該版本首次擴展至Windows系統與Radeon消費級顯示卡,集成企業級MLOps平台和Hugging Face模型,支持AMD Instinct MI350系列GPU。硬體支持涵蓋CDNA4架構的Instinct MI350系列GPU,提供288GB HBM3E記憶體支持,並通過AMD雲服務提供MI系列GPU資源。
2025年9月25日發布的ROCm 6.4.4預覽版實現Windows系統PyTorch框架全面支持,覆蓋Radeon RX 9000/7000系列顯示卡及銳龍AI系列APU。
基本介紹
- 最近更新時間:2025年4月
- 開發商:AMD
- 軟體版本:6.4.0
- 外文名:ROCm
- 產品類型:GPU計算生態系統
發展歷程
主要功能
版本記錄
- ROCm 版 Megatron-LM 框架新增融合注意力、融合層範數、融合旋轉式位置編碼 RoPE 核心支持;
- 支持 PyTorch 2.6 和 2.5 版本;
- rocDecode 和 rocPyDecode 支持 VP9 編解碼;
- 對於 Instinct 加速器支持 Oracle Linux 9 作業系統。

