PyTorch终于能用上谷歌云TPU，推理性能提升4倍，该如何薅羊毛？(2)

发布时间：2019-10-14 14:28 所属栏目：19 来源：晓查

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

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