TensorFlow为新旧Mac..新版本，GPU可用于训练，速度最高提升7倍

2020-11-20 15:23:55

机械之心报道

[好文分享：www.ii77.com]

机械之心编纂部

苹果「一呼百诺」的号召力在机械进修范畴似乎也不破例。新版 Mac 推出还不到两周，谷歌就把专为 Mac 优化的 TensorFlow 版本做好了，练习速度最高提拔到本来的 7 倍。

对于斥地者、工程师、科研工作者来说，Mac 一向是非常受迎接的..，也有人用 Mac 练习神经收集，但练习速度一向是一个令人头疼的问题。

[原文来自：www.ii77.com]

上周，苹果发布了搭载 Arm 架构 M1 芯片的三款新 Mac，于是就有人想问：用它们练习神经收集能快一点吗？

今天，主流机械进修框架 TensorFlow 发文透露：我们专门做了一版为 Mac 用户优化的 TensorFlow 2.4 框架，M1 版 Mac 和英特尔版 Mac 都能用。这一勾当有望大幅降低模型练习和布置的门槛。

此前，在 Mac 上，TensorFlow 仅支撑将 CPU 用于练习，但新的 tensorflow_macos 分支行使苹果的 ML Compute，能让 GPU 也被行使起来。苹果在博客中介绍说：「我们使用了更高级其余优化方式，好比熔合层，选择合适的设备类型，将图作为原语编译、执行并由 CPU 上的 BNNS 和 GPU 上的 Metal Performance Shader 加快。」

ML Compute 是苹果公司本年推出的新框架，可用于在 Mac 长进行 TensorFlow 模型的练习。如今，无论新的 M1 版 Mac 照样旧的英特尔版 Mac，其 CPU 和 GPU 都能用来加速练习速度。

M1 芯片包含新的 8 核 CPU 和最多 8 核的 GPU，并针对 Mac 的机械进修练习义务进行了优化。下面两张图表离别展示了针对 Mac 优化后的 TensorFlow 2.4 在分歧模型练习中的机能提拔：

上图展示了使用 ML Compute 离别在搭载 M1 和英特尔芯片的 13 英寸 MacBook Pro 长进行机械进修练习的情形。柱形的高度代表单批次的练习时长。能够看出，优化后的 Mac 版 TensorFlow 2.4 的练习速度获得了显着提拔。在换成 M1 版的新 MacBook Pro 之后，提拔就更为显着了，练习速度最高提拔了 7 倍。

同样使用 ML Compute，在搭载英特尔芯片的 2019 Mac Pro 长进行常见模型的练习，机能提拔也相当显着。

TensorFlow 官方透露，用户不需要对现有的 TensorFlow 剧本做任何更改，就能够使用 ML Compute 作为 TensorFlow 和 TensorFlow 插件的后端。

Mac 版 TensorFlow 2.4 的具体入门指南能够拜见：https://github.com/apple/tensorflow_macos

M1 版 Mac 对机械进修用户的影响

苹果曾在发布会上介绍说，M1 芯片将苹果的神经收集引擎引入了 Mac，实现了 15 倍的机械进修义务加快。该神经收集引擎有 16 个焦点，每秒运算速度可达 11 万亿次。除此之外，设置了 ML 加快器的 CPU 和壮大 GPU（称霸集显，媲美部门独显）也使得整个 M1 芯片的机械进修能力获得伟大提拔。

图源：https://medium.com/disruptive-nerd/how-is-the-apple-m1-going-to-affect-machine-learning-2d9da1beef86

机能的提拔毋庸置疑，但软件兼容也是一个很大的问题。在 M1 版 Mac 发布之后，不少斥地者都在问：我的 VS Code/Docker/Juypter Notebook…… 还能在新 Mac 上用吗？

VSCode：最快月底就能做出来

在苹果发布会之后不久，微软 VS Code 团队就发文透露，「VS Code 将支撑 ARM 版 Mac，11 月底或者推出首个预览版。」

Mac/ARM64 试验版地址 https://code.visualstudio.com/insiders/#osx

此外，微软还公布将发布一款新的通用版 Mac Office 2019，支撑新版苹果处理器。

Docker：在放置，但还没预备好

在 11 月 16 日的一篇博客中，Docker 的斥地者透露，他们还没有预备好让 Docker 在 M1 版 Mac 上运行，在这之前还需要做好多预备。

Docker 透露，「我们 Docker 的方针是在新的 Mac 上供应与今天在 Docker Desktop for Mac 上同样超卓的体验，并使这种过渡尽或者地无缝。」「为我们的客户打造合适的体验意味着在我们推送一个版本之前，要把不少事情做好。固然苹果已经发布了 Rosetta 2 转译器，以匡助将应用法式转移到新的 M1 芯片上，但这并不克让我们与 Docker Desktop 一路同业。」

至于 Juypter Notebook 等软件，今朝还没有官方新闻放出。不外，跟着苹果络续将生态打通，正在有越来越多的软件斥地商透露将斥地适配 M1 版 Mac 的新产物。

参考链接：

https://machinelearning.apple.com/updates/ml-compute-training-on-mac

https://medium.com/disruptive-nerd/how-is-the-apple-m1-going-to-affect-machine-learning-2d9da1beef86

https://www.docker.com/blog/apple-silicon-m1-chips-and-docker/

11 月 21 日，华为云DevRun斥地者沙龙将上岸武汉。

我们预备了2场关于CloudIDE和API开放..的干货分享和60分钟聚焦华为云API入门实践的面临面切磋。届时，华为云高级产物司理赵彦、华为云生态高级软件工程师胡建将经由自身工作实践，一一解答关于云原生、高效斥地、集成进修等问题。

识别海报..，立刻报名。

转载请关联本公家号获得授权

投稿或追求报道：content@jiqizhixin.com

上一篇：超高清！视频可以这样看
下一篇：绕过VPN和防火墙？苹果给自家APP留后门

TensorFlow为新旧Mac..新版本，GPU可用于训练，速度最高提升7倍

热门文章

小编推荐

推荐文章