已针对 AI 工作负载进行优化

AMD ROCm 是一个开源软件堆栈，面向 AI 工作负载提供一系列优化方案，并支持庞大的 AI 软件生态系统，包括多种开放式框架、模型及工具。

简化的模型开发

支持 Hugging Face 上的多种模型，可用于开发满足用户特定需求的解决方案

支持主流先进框架

针对 TensorFlow 和 PyTorch 等主流先进 AI 框架提供上游支持

机器学习库

提供有关机器学习的专有库，其中包括 MIOpen 和 MIVisionX 等

横向扩展部署

支持 Docker、Singularity、Kubernetes® 和 Slurm 等主要容器化工具，以实现大规模部署

最新动态

了解有关使用 ROCm 进行 AI 开发的最新进展。

查看全部

ROCm 助力人工智能 — 开源软件堆栈

AI 解决方案和服务

  • 计算机编程
  • 内容生成
  • 客户服务聊天机器人
  • 数据分析
  • 开发者工具
  • 员工助手
  • 知识管理
  • 流程自动化
  • 推荐系统
  • 报告生成
  • 总结

  • Bloom
  • Falcon 大语言模型
  • GPT-4
  • Llama 2
  • Llama 3
  • Mistral AI
  • Stable Diffusion
  • StarCoder
  • Qwen
  • Yi

AI 生态系统

ROCm AI partner logos

GPU

AMD Instinct white logo
AMD Radeon white logo
Programming code abstract technology background of software developer and Computer script

调用全新 AI 模型

AMD Infinity Hub 包含一系列高级软件容器和 AI 应用部署指南。

访问 Infinity Hub

AMD Developer Cloud 助力构建 AI 模型

申请体验 AMD Instinct 加速器以简化开发过程。

立即申请
AMD Data Center and Cloud Computing

合作伙伴和成功案例

了解 AMD 如何与各大企业和研究机构携手推进 AI 发展。

Lamini logo

Lamini

了解 Lamini 软件和 ROCm 如何为 finetuning LLM 等高性能应用奠定坚实的基础。

阅读博客
Mosaic ML logo

Mosaic ML

了解 Mosaic ML 在不更改代码且保持系统高效运行的情况下，如何利用 ROCm 通过 AMD Instinct 加速器开展 LLM 训练。

阅读博客
Hugging Face logo

Hugging Face

了解 Hugging Face 和 AMD 如何携手加速先进模型，以及 Hugging Face 如何将 ROCm 集成到其开源库中。

阅读博客
Pytorch logo

PyTorch

AMD 正与 PyTorch Foundation 携手合作，打造开放式 AI 软件生态系统，并致力于增加 ROCm 对 PyTorch 的支持。

阅读博客

支持的 GPU

开发人员资源

ROCm Developer screen shot

ROCm 开发人员中心

着手开发支持 AMD GPU 加速的应用。访问 ROCm 开发人员中心，获取最新的用户指南、容器、培训视频、网络研讨会等内容。

立即体验

基于 AMD Radeon GPU 进行 ML 开发

使用 PyTorch 处理 ML 模型与算法时，研究人员和开发人员现在还可以在 Ubuntu® Linux® 上使用 ROCm（从 5.7 版开始），充分发挥全新 AMD Radeon 7900 系列台式机 GPU 强大的并行计算能力。²

立即体验

ROCm 新闻通讯

接收最新 ROCm 资讯。

随时掌握最新动态
附注

©2024 AMD 公司，版权所有。保留所有权利。AMD、AMD 箭头标识、AMD ROCm、AMD Instinct、EPYC（霄龙）、Radeon Instinct 及其组合是 AMD 公司的商标。PyTorch 是 PyTorch 公司的商标或注册商标。本文中用到的其他产品名称仅用于标识目的，可能是其各自公司的商标。

  1. 如需了解 ROCm 所支持的全部 Radeon 产品，请访问 https://rocm.docs.amd.com/en/latest/reference/gpu-arch-specs.html
  2. Radeon AI 技术兼容所有 AMD Radeon 7000 系列及后续推出的显卡。购买前请先咨询您的系统制造商是否支持相关功能。GD-232。