麦克雷

标题: PyTorch 将支持在苹果芯片 Mac 上开启 GPU 加速 [打印本页]

作者: 上海秘书 时间: 2022-5-21 15:57
标题: PyTorch 将支持在苹果芯片 Mac 上开启 GPU 加速

PyTorch 今天宣布，与苹果 Metal 工程团队合作后，PyTorch 开源机器学习框架将很快支持在苹果芯片 Mac（包括 M1、M1 Pro、M1 Max 和 M1 Ultra 芯片）上开启 GPU 加速模型训练。

(, 下载次数: 10)

目前，Mac 上的 PyTorch 训练仅调用 CPU，未来版本的 PyTorch 将可以在苹果芯片 Mac 上开启 GPU 加速，性能更快。

欢迎光临麦克雷 (http://imac.ly/)