加入收藏 | 设为首页 | 会员中心 | 我要投稿 湖南网 (https://www.hunanwang.cn/)- 科技、建站、经验、云计算、5G、大数据,站长网!
当前位置: 首页 > 建站 > 正文

PyTorch终于能用上谷歌云TPU,推理机能晋升4倍,该怎样薅羊毛?

发布时间:2019-10-15 10:12:15 所属栏目:建站 来源:晓查
导读:本文经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

(编辑:湖南网)

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

热点阅读