网易首页 > 网易号 > 正文 申请入驻

PyTorch 1.9发布!移动端疯狂更新,网友:我的最爱

0
分享至

明敏 发自 凹非寺
量子位 报道 | 公众号 QbitAI

时隔仅3个月,PyTorch再次迎来升级——1.9版本

这一次,官方把重头戏放在了移动端上。

不仅Mobile Interpreter发布了新版本,而且TorchVision库也支持在手机上使用了,iOS、Android都支持!


网友看了都表示:

这一次更新中,我对移动端最感兴趣。


而此次1.9版本集合了自2021年3月1.8版本发布以来,超过3400次GitHub提交。

除了移动设备端方面,还有其他诸多亮点:

前端API改进(包括torch.linalg、torch.special和 Complex Autograd)

实现对弹性、容错分布式训练的本地支持

更新PyTorch模型性能分析器

LeCun也在Twitter上站台,称赞PyTorch做的一次比一次好!


首先就是PyTorch Mobile最受欢迎的功能之一Mobile Interpreter更新了。

最新版本能够将移动设备上的二进制文件大小降低到原来大小的一半以下

比如,在arm64-v8a架构的Android设备中使用MobileNetV2的pt大小,压缩前为17.8MB、压缩后为8.6MB。

而使用新版Mobile Interpreter后,可以把压缩前的文件大小降低到8MB以下,压缩后的大小降低到4MB以下。

与此同时,从1.9版本开始,用户可以在iOS、Android的APP上使用TorchVision库了。

在iOS上,它需要和主要的PyTorch库链接在一起使用;在Android上,则可作为gradle依赖项添加。

在demo APP方面,这一次更新了一个新的基于PyTorch Video库的视频APP和一个基于最新torchaudio,、wave2vec模型的语音识别APP。

有了这两个APP后,PyTorch现在可以提供图像、文字、音频和视频在内的一整套demo APP。


1.9版本中,对torch.linalg、torch.special和Complex Autograd等模块进行了完善。

torch.linalg模块现在可以实现NumPy线性代数模块中的每个函数;

Complex Autograd更新的新功能,可以计算复杂梯度、优化损失函数。

此外,为了帮助调试和编写可重现的程序,PyTorch 1.9增加了一个torch.use_determinstic_algorithms选项。

这是为了避免运行中可能会出现的错误,如下所示:


TorchElastic是PyTorch的一个核心功能,它能够让用户在抢占式实例上运行分布式训练。


△TorchElastic运行原理

新版本中,添加了一个基于c10d::Store的“独立”集合点,能够在本地支持弹性、容错分布式训练。

此外,现在还可在RPC中支持CUDA,并支持对分布式训练进行分析等。

PyTorch Profiler是用来对PyTorch模型性能分析的一个工具。

通过可视化页面来帮助我们分析具体运行情况。


在1.9版本中,可以在Windows和Mac中支持新的torch.profiler API了。

新的API支持现有分析器功能,可以与CUPTI库集成(仅限Linux),追踪设备上的CUDA内核,并为长期运行项目提供支持,例如:


PyTorch Profiler Tensorboard插件还更新了带有NCCL的分布式训练摘要视图、内存分析视图、从Microsoft VSCode启动时可跳转到源代码等新功能。

特别声明:以上内容(如有图片或视频亦包括在内)为自媒体平台“网易号”用户上传并发布,本平台仅提供信息存储服务。

Notice: The content above (including the pictures and videos if any) is uploaded and posted by a user of NetEase Hao, which is a social media platform and only provides information storage services.

相关推荐
热点推荐
爆冷!国羽女单逆转日本队头号选手!裁判引发争议,非常尽力了

爆冷!国羽女单逆转日本队头号选手!裁判引发争议,非常尽力了

重新出发的体娱人
2021-07-30 09:41:31
上海市委常委、浦东新区区委书记翁祖亮,任中国五矿董事长、党组书记

上海市委常委、浦东新区区委书记翁祖亮,任中国五矿董事长、党组书记

新京报政事儿
2021-07-30 13:53:10
中国女排实已无缘出线,将破29年最差纪录,郎平谢幕方式令人心碎

中国女排实已无缘出线,将破29年最差纪录,郎平谢幕方式令人心碎

毛罗大神
2021-07-29 18:36:50
美国公开天津会谈内容,谢尔曼直接给中国定罪,为何要激怒中方?

美国公开天津会谈内容,谢尔曼直接给中国定罪,为何要激怒中方?

刘晓非说
2021-07-29 15:14:53
伊藤美诚,既可恨,又可怜!

伊藤美诚,既可恨,又可怜!

辣笔小尖椒
2021-07-29 19:16:42
焦点分析 当P50不再有5G版本,华为卖起了儿童学习机

焦点分析 当P50不再有5G版本,华为卖起了儿童学习机

36氪
2021-07-30 08:40:28
俄罗斯科学号实验舱发动机突然自启动,使国际空间站旋转 45 度

俄罗斯科学号实验舱发动机突然自启动,使国际空间站旋转 45 度

IT之家
2021-07-30 09:24:31
何诗蓓再次打破100米自由泳亚洲纪录

何诗蓓再次打破100米自由泳亚洲纪录

乐活体育圈
2021-07-30 10:35:20
连体裤上厕所方便吗?

连体裤上厕所方便吗?

火火
2021-07-29 22:57:22
女护士因欠巨款,与男医生发生多次性行为,最后将其杀死肢解

女护士因欠巨款,与男医生发生多次性行为,最后将其杀死肢解

关注娱乐的秀
2021-07-28 15:01:24
朱婷究竟是怎么受伤的?女排助教说出真相,彻底点燃球迷怒火

朱婷究竟是怎么受伤的?女排助教说出真相,彻底点燃球迷怒火

大咖唠体育
2021-07-29 11:01:44
8月起,新冠疫苗接种有变

8月起,新冠疫苗接种有变

中外医讯
2021-07-29 10:23:35
向太官宣儿媳怀孕消息,郭碧婷怀男怀女惹热议,粉丝:终于怀上了

向太官宣儿媳怀孕消息,郭碧婷怀男怀女惹热议,粉丝:终于怀上了

文文趣说娱乐
2021-07-30 13:39:54
“水跟方便面真的不要再捐了”说明了什么

“水跟方便面真的不要再捐了”说明了什么

法制社会报
2021-07-29 17:24:20
西班牙王室那个可怕的孩子长大了,就连国王都忍不住让姐姐管管他

西班牙王室那个可怕的孩子长大了,就连国王都忍不住让姐姐管管他

要闻每日关注
2021-07-30 09:22:11
郑州地铁遇难名单公布,生前遗言戳痛无数人“妈妈,不要挂念我”

郑州地铁遇难名单公布,生前遗言戳痛无数人“妈妈,不要挂念我”

深後
2021-07-29 16:29:50
特斯拉又降价!Model 3价格下调1.5万元,此前Model Y直降7.19万

特斯拉又降价!Model 3价格下调1.5万元,此前Model Y直降7.19万

和讯网
2021-07-30 11:20:43
南京正在进行第三轮核酸采样,已查出15例阳性

南京正在进行第三轮核酸采样,已查出15例阳性

澎湃新闻
2021-07-30 11:19:24
伊藤美诚又哭了!获取铜牌后吐露狠话!和别的选手真不一样

伊藤美诚又哭了!获取铜牌后吐露狠话!和别的选手真不一样

运动帮
2021-07-29 22:18:02
中国游泳队是拯救了银河系吗?完全沦陷在他们的颜值里!

中国游泳队是拯救了银河系吗?完全沦陷在他们的颜值里!

摄影技巧入门教程
2021-07-29 17:32:55
2021-07-30 14:54:44
量子位
量子位
追踪人工智能动态
5236文章数 119006关注度
往期回顾 全部

科技要闻

特斯拉Model 3国内起售价下调至23.59万元

头条要闻

美国在奥运男子200米混的17年垄断 由汪顺打破

头条要闻

美国在奥运男子200米混的17年垄断 由汪顺打破

体育要闻

超燃!汪顺5年让奖牌换色 中国泳坛一哥

娱乐要闻

脸垮被嘲 失去美貌的热巴还剩什么?

财经要闻

汽车要闻

先立个flag再说 保时捷发布六座MPV概念车

态度原创

时尚
亲子
手机
本地
艺术

昨天LV舍弃吴亦凡 今天LVMH"吞食"了他的好兄弟

亲子要闻

暑假作业敷衍、拖拉的毛病,今年确定改得了吗?

手机要闻

华为P50 Pro体验:众人期待值拉满后,它终于来了

本地新闻

除了木村拓哉,日本到底还有帅哥吗?

艺术要闻

后印象派的风景油画,画风柔美亲切