windsing 在 谷歌推出了 TorchTPU,可让 AI 应用在 TPU 和谷歌云上运行 中发帖

谷歌与亚马逊一同向英伟达的阵地发起了进攻。谷歌推出了 TorchTPU,这是一个开源版本的 PyTorch,旨在让 AI 应用程序通过简单的配置更改即可在 TPU 和谷歌云上运行,而无需重新编写代码。 
Google 推出的 TorchTPU 并不是重新发明一个封闭的 PyTorch,而是基于官方 PyTorch 代码库,深度定制并整合了 Google TPU 底层编译器(XLA/Pallas)的开源衍生版本 
过去,全球绝大多数 AI 算法工程师写的 PyTorch 代码,默认调用的都是 torch.cuda、底层跑的都是英伟达的 cuDNN/TensorRT 算子。 
谷歌以前也做过 PyTorch/XLA 兼容层,但配置繁琐、编译慢、经常遇到算子不兼容报错,开发者往往需要大量重写模型训练代码才能在 TPU 上跑起来。这导致很多开发者即使觉得谷歌 TPU 便宜算力高,也懒得迁移。 
G...
 
 
Back to Top