Apple释出专为Mac最佳化的TensorFlow版本

Apple释出专为Mac最佳化的TensorFlow 2.4版本，经实验证实，使用最佳化的版本，无论是在MacBook Pro还是Mac Pro上，训练工作负载中每批次执行时间，都比使用TensorFlow 2.3还要短，而且在使用Apple自家M1芯片的Mac，执行TensorFlow 2.4工作负载，效能要比使用英特尔芯片的Mac高出不少。

深度学习函式库TensorFlow 2，可让开发者在不同的平台、装置和硬件上，训练机器学习模型，而Apple现在专门针对Mac，释出最佳化TensorFlow版本和新的机器学习运算框架（ML Compute Framework），使得Mac上模型训练工作能够更加快速。

新加入的机器学习运算框架，可以利用Mac的硬件，来加速模型训练工作负载，无论是使用搭载M1和英特尔芯片的Mac，都能获得效能提升，官方提到，特别是M1芯片中拥有高达8核CPU和8核GPU，更可极大程度提升训练效能。

M系列芯片为Apple Silicon其中一个系列，从iPhone 4开始，Apple就开始采用基于Arm架构，自己设计的A4芯片，此后包括AirPods、Apple Watch和HomePod，也都是使用Apple芯片，而在2020年6月，Apple首席执行官Tim Cook在Apple全球开发者大会上宣布，启动Mac芯片搬迁计划，Mac系列电脑在两年内，将从英特尔公司的处理器，改为使用Apple自己设计的芯片。

在11月刚结束不久的线上发表会，Apple发表了M系列芯片，以及该系列第一款芯片M1，同时也推出新款使用M1芯片的Mac产品。M1除了拥有多颗运算核心之外，还在单一芯片上整合多种技术，且具备统一内存架构，可大幅改善效能，是目前第一款以5奈米制程生产的个人电脑芯片，而这几天使用M1芯片的新Mac评测不断出炉，无论是在效能还是电池续航力，都获得了极高的评价。

官方使用TensorFlow 2.4进行实验，发现TensorFlow 2.4在Mac上效能，的确较TensorFlow 2.3好，而M1芯片的表现也比英特尔芯片还要好。结果显示M1芯片在机器学习领域，似乎也有很大的发展空间。

在2020最新使用英特尔芯片的MacBook Pro上，执行TensorFlow 2.3和经最佳化的TensorFlow 2.4，在Style Transfer以及DenseNet121等工作负载，TensorFlow 2.4版本每批次执行秒数都明显较低，而在使用M1芯片的MacBook Pro执行TensorFlow 2.4，比起英特尔的MacBook Pro的效能，更是快上好几倍。

在Mac Pro的比较上，由于尚未有M1芯片的机型，因此是以2019年使用英特尔芯片的Mac Pro，分别执行TensorFlow 2.3和经最佳化的TensorFlow 2.4进行比较，而结果也显示TensorFlow 2.4带来大量的效能提升，甚至在DenseNet121等模型，效能提升超过10倍。

官方提到，开发者不需要更改现有TensorFlow脚本，就能使用新加入的机器学习运算框架，机器学习运算框架会以TensorFlow的后端，或是扩充套件的形式运作。开发者现在就可以在Apple的GitHub储存库中，下载Mac最佳化的TensorFlow 2.4分支，之后TensorFlow官方会在主要分支中，合并Apple的版本，开发者将更容易获取更新和使用，为Mac最佳化的功能。