元峰 · 2022年10月17日 · 北京市

苹果 M1 芯片,跑深度学习到底有多快?

苹果的笔记本,现在基本都要替换为自研的 M1 CPU了。据说,M1 跑深度学习也很快,那到底有多快呢。

  

先看一张图。跑知名的 VGG 16 模型,M1 Pro 的 CPU 是比英特尔的 E5-2650 (服务器 CPU)还要猛的,M1 Pro GPU更是比英特尔快一倍多。但是,跟 NVIDIA 的显卡还是有很大差异的。

  
image.png
  

最近,PyTorch 官方也宣布,新版本的 PyTorch 也要支持 M1 GPU 训练了,并且给出了 GPU 相比 CPU 的加速倍数。从下图可以看出,推理上, M1 ultra 的 GPU,比 CPU 要快十几倍。

640.gif

图中黑线是 CPU 基准速度,紫色和红色是 GPU 相比 CPU 的速度。  

  

可以看出,苹果的 M1系列 GPU,相比自身的 CPU 还是有很大优势的,比英特尔的 CPU 也要强很多。  

作者:AIZOO
文章来源:AIZOO 小可爱

推荐阅读

更多嵌入式AI干货请关注 嵌入式AI 专栏。欢迎添加极术小姐姐微信(id:aijishu20)加入技术交流群,请备注研究方向。
  
推荐阅读
关注数
225
内容数
48
AIZOO.com 致力于搭建AI开发者、AI公司与需求方的桥梁,打造中国最大的算法和产品商城。传播AI领域资讯和技术,展示AI算法和产品。
目录
极术微信服务号
关注极术微信号
实时接收点赞提醒和评论通知
安谋科技学堂公众号
关注安谋科技学堂
实时获取安谋科技及 Arm 教学资源
安谋科技招聘公众号
关注安谋科技招聘
实时获取安谋科技中国职位信息