PyTorch终于支持苹果的M1芯片了！

自从苹果发布M1系列的自研芯片开始，基于ARM架构的电脑处理器开始大放异彩。而强大的M1芯片的能力也让很多Mac用户高兴很久。而就在现在，M1也开始支持PyTorch的深度学习框架了。PyTorch官网刚刚宣布，经过和Apple的Metal工程师队伍的合作，PyTorch支持Mac的GPU加速了。

使用苹果公司的Metal Performance Shaders（MPS）作为PyTorch的后端，可以实现加速GPU训练。MPS后端扩展了PyTorch框架，提供了在Mac上设置和运行操作的脚本和功能。MPS通过针对每个Metal GPU系列的独特特性进行微调的内核来优化计算性能。新设备在MPS图形框架和MPS提供的调整内核上映射机器学习计算图形和基元。

如上图所示，相比较CPU训练和评估而言，GPU的训练速度明显快很多。

不过，可惜的是并不是所有的PyTorch操作目前都可以支持，例如flip、indexing操作貌似都不行。不过，不管怎么说，如果继续合作的化，那么以后PyTorch可能会在苹果软硬件上有更好的表现了。

其实，与X86架构的处理器相比，ARM本身在图形处理、日常工作中的能力早就可以支撑很好了。不过，由于Windows等生产力工具的限制，尽管CPU的硬件性能没问题，但是缺乏好的生态依然难以让ARM在很多领域发挥价值。包括华为之前发布的基于ARM芯片的电脑，如果无法将操作系统和软件生态一起打包让消费者接受，其实很难获取成功的。