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倍阁下的晋升。 ![]() GitHub地点: https://github.com/pytorch/xla/tree/master/contrib/colab
(编辑:湖南网) 【声明】本站内容均来自网络,其相关言论仅代表作者个人观点,不代表本站立场。若无意侵犯到您的权利,请及时与联系站长删除相关内容! |