加入收藏 | 设为首页 | 会员中心 | 我要投稿 云计算网_泰州站长网 (http://www.0523zz.com/)- 视觉智能、AI应用、CDN、行业物联网、智能数字人!
当前位置: 首页 > 运营中心 > 建站资源 > 经验 > 正文

PyTorch终于能用上谷歌云TPU,推理性能提升4倍,该如何薅羊毛?

发布时间:2019-10-15 15:35:03 所属栏目:经验 来源:晓查
导读:副标题#e# 本文经AI新媒体量子位(公众号ID:QbitAI)授权转载,转载请联系出处。 Facebook在PyTorch开发者大会上正式推出了PyTorch 1.3,并宣布了对谷歌云TPU的全面支持,而且还可以在Colab中调用云TPU。 之前机器学习开发者虽然也能在Colab中使用PyTorch,

在PyTorch上使用TPU对性能的提升到底有多明显呢?官方选用了v2-8的一个核心,即1/8 TPU的情形,与使用英伟达Tesla K80 GPU进行对比,实测显示推理时间大大缩短,性能约有4倍左右的提升。

PyTorch终于能用上谷歌云TPU,推理性能提升4倍,该如何薅羊毛?

GitHub地址:

https://github.com/pytorch/xla/tree/master/contrib/colab

(编辑:云计算网_泰州站长网)

【声明】本站内容均来自网络,其相关言论仅代表作者个人观点,不代表本站立场。若无意侵犯到您的权利,请及时与联系站长删除相关内容!

热点阅读