ROCm 是一个开源软件平台，允许研究人员利用 AMD Instinct 加速器的潜力，促进跨平台的高性能计算和 AI 创新。

AMD 上个月宣布为基于 RDNA 3 的 Radeon Pro W7900 和 Radeon RX 7900 XTX 显卡带来了对 PyTorch 2.0.1 的 ROCm 支持，今天正式将其兼容范围扩展到 RX 7900 XT 次旗舰游戏显卡上。

AMD Radeon RX 7900 XT 基于 Navi 31 GPU 芯片，拥有 20 GB 超大显存以及总共 168 个 AI 加速器单元，支持 ML 训练和推理加速。

据介绍，该 GPU 还支持基于 PyTorch 的 ML 模型和算法，这些模型和算法已经专门针对 AMD 进行了优化，并将与运行 22.04.3 版本（Ubuntu）的 Linux 操作系统兼容，感兴趣的IT之家小伙伴可以前往官网下载。

这一举措使开发人员能够使用配备 AMD 消费级旗舰 GPU 利用 PyTorch 库进行人工智能应用的开发，从而避免对于英伟达的过度依赖。

此外，该驱动专门针对 PyTorch 2.0.1 进行了优化，适配了 AMD RDNA3-based GPU 的功能。

本文转载自IT之家，转载目的在于传递更多信息，并不代表本站赞同其观点和对其真实性负责。如涉及作品内容、版权和其它问题，请联系IT之家通知我方删除，我方将在收到通知后第一时间删除内容！本文只提供参考并不构成任何投资及应用建议。本站拥有对此声明的最终解释权。