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 操作系统兼容,感兴趣的小伙伴可以前往官网下载。
这一举措使开发人员能够使用配备 AMD 消费级旗舰 GPU 利用 PyTorch 库进行人工智能应用的开发,从而避免对于英伟达的过度依赖。
此外,该驱动专门针对 PyTorch 2.0.1 进行了优化,适配了 AMD RDNA3-based GPU 的功能。