谷歌开源PyTorch版本TorchTPU,开发者可迁移AI应用至TPU及谷歌云
2026年10月20日,谷歌宣布其长期开发的PyTorch版本TorchTPU正式以开源项目形式发布。这一版本允许AI应用在谷歌自研的张量处理单元(TPU)芯片上运行,开发者通过简单的配置变更即可将现有PyTorch应用迁移至TPU及谷歌云平台,无需重写代码。
TorchTPU是谷歌为PyTorch这一被广泛使用的AI开发框架打造的自研芯片适配版本。PyTorch最初由Meta于2016年开发,2022年移交给Linux基金会管理,是AI开发者构建、训练和运行AI模型的主流开源工具。谷歌表示,其已与Meta、Hugging Face、Inferact、RadixArk等机构合作,确保TorchTPU的功能和代码库与这些常用AI开发工具兼容。
此外,谷歌还推出了专用的TPU agent suite工具,帮助开发者将应用迁移至谷歌芯片和云平台,并处理迁移过程中的错误。谷歌还更新了其LiteRT工具,使运行在手机、笔记本电脑和嵌入式设备上的AI应用能更好地兼容PyTorch。
在谷歌发布TorchTPU之前,亚马逊于2026年10月19日发布了其自研PyTorch版本TorchNeuron,允许AI应用无需重写即可迁移至其Trainium芯片和AWS平台,并已向PyTorch主项目提交代码,以推动PyTorch与英伟达芯片解耦。谷歌和亚马逊的举动被视为对英伟达在AI芯片领域主导地位的挑战。
- 2026-10-21 02:38 | TechCrunch:Google has joined Amazon in attacking Nvidia’s stronghold with AI app developers阅读原文
On Tuesday, Google announced that TorchTPU, its long-awaited version of PyTorch, is now available as an open-source project. With this, Google joins Amazon in trying to break Nvidia’s stranglehold on the AI industry, where most AI apps are written by default to run on Nvidia’s chips, or GPUs. PyTor...