谷歌 TensorFlow 2.4 Mac M1 优化版发布:相比 Intel MacBook Pro/Mac Pro 性能巨大提升

2020-12-03 20:16IT之家 - 骑士

IT之家12月3日消息 谷歌表示,借助 TensorFlow 2,可在跨平台、设备和硬件上实现一流的训练性能,从而使开发者、工程师和研究人员能够在他们喜欢的平台上工作。IT之家获悉,现在,TensorFlow 用户可以在搭载 Apple 全新 M1 芯片或 Intel 芯片 Mac 上的 利用 TensorFlow 2.4 Mac 优化版和新的 ML Compute 框架来加快训练速度。这些改进提升了 Apple 开发者通过 TensorFlow Lite 在 iOS 上执行 TensorFlow 的能力,继续展现了 TensorFlow 在 Apple 硬件上支持高性能 ML 执行方面的广度和深度。

采用 ML Compute 时 Mac 上的性能

Apple 近期发布了搭载全新 M1 芯片的系列 Mac 产品,如此一来,Apple 针对 Mac 优化的 TensorFlow 2.4 版能够充分利用 Mac 的强大功能并在性能上大幅提升。

ML Compute 是 Apple 的新框架,可以在 Mac 上训练 TensorFlow 模型,现在,您可以在搭载 M1 和 Intel 芯片的 Mac 上实现加速的 CPU 和 GPU 训练。

例如,M1 芯片搭载功能强大的新型 8 核 CPU 和多达 8 核 GPU,均针对 Mac 上的 ML 训练任务进行了优化。在下图中,您可以看到针对 Mac 优化的 TensorFlow 2.4 如何在搭载 M1 和 Intel 芯片的通用型号 Mac 上实现巨大的性能提升。

▲在搭载 M1 和 Intel 芯片的 13 英寸 MacBook Pro 上使用 ML Compute 时对常见模型训练影响,以每批秒数显示,数字越小表示训练时间越短

▲在搭载 Intel 芯片的 2019 Mac Pro 上使用 ML Compute 时对常见模型的训练影响,以每批秒数显示,数字越小表示训练时间越短

开始使用针对 Mac 优化的 TensorFlow

用户无需对其现有的 TensorFlow 脚本进行任何更改即可使用 ML Compute 用作 TensorFlow 和 TensorFlow 插件的后端。

首先,请访问 Apple 的 GitHub 仓库,了解如何下载和安装 Mac 优化的 TensorFlow 2.4。

在不久的将来,谷歌会将该版本集成到 TensorFlow master 分支中,使用户能更轻松地进行此类更新,从而获得这些性能数据。

您可以在 Apple 的机器学习网站上了解 ML Compute 框架细节。

广告声明:文内含有的对外跳转链接(包括不限于超链接、二维码、口令等形式),用于传递更多信息,节省甄选时间,结果仅供参考,IT之家所有文章均包含本声明。

文章价值:
人打分
有价值还可以无价值
置顶评论
    热门评论
      全部评论
      一大波评论正在路上
        取消发送
        软媒旗下人气应用

        如点击保存海报无效,请长按图片进行保存分享