RX7900GRE支持哪些AI框架

rx7900gre的ai框架支持

• pytorch（rocmin 6.0 起）：支持在 rx 7900 gre 上进行机器学习训练，提供 fp32/fp16 混合精度 训练工作流。适合从研究到生产的深度学习训练与推理场景。
• onnx runtime（rocmin 6.0 起）：通过 migraphx 图推理引擎在本地 amd 硬件上运行模型推理，支持的数据类型包括 fp32、fp16、int8，便于跨框架模型部署与加速。
• tensorflow（rocmin 6.1.3 起）：已获得 rocm 认证，rx 7900 gre 在该版本起可原生用于 tensorflow 的 ai 加速开发，与 pytorch、onnx 共同覆盖主流框架生态。

运行环境与版本要点

• 操作系统与平台：rocmin 6.1.3 的 radeon software for linux 兼容 ubuntu 22.04.3 hwe（64 位）；同时在 rocmin 6.1.3 中新增了对 wsl 2（windows subsystem for linux）beta 支持，便于在 windows 上运行基于 linux 的 ai 工具链。
• 典型使用路径：在 pytorch 中启用 rocm 后端进行训练；将模型导出为 onnx 并在 onnx runtime（migraphx） 上做高性能推理，必要时使用 int8 量化以进一步提升吞吐与能效。