找回密码
 立即注册
  • QQ空间
  • 回复
  • 收藏

新闻 PyTorch 将支持在苹果芯片 Mac 上开启 GPU 加速


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

225355boo88die08ri8i8o.jpg

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

使用道具 举报

说点什么

您需要登录后才可以回帖 登录 | 立即注册
HOT • 推荐