PyTorch终于能用上谷歌云TPU，推理性能提升4倍，该如何薅羊毛？
发布时间：2019-10-19 06:29:26 所属栏目：经验 来源：晓查
导读：副标题#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倍左右的提升。 
GitHub地址： https://github.com/pytorch/xla/tree/master/contrib/colab
（编辑：宁德站长网） 【声明】本站内容均来自网络，其相关言论仅代表作者个人观点，不代表本站立场。若无意侵犯到您的权利，请及时与联系站长删除相关内容! |
