谷歌开源 TorchTPU，挑战英伟达 AI 生态主导地位

谷歌当地时间周二宣布，面向张量处理单元（TPU）的原生PyTorch框架TorchTPU已在GitHub开源。该工具可让人工智能应用在谷歌自研芯片上运行，被视为挑战英伟达行业主导地位的举措。目前，绝大多数人工智能应用默认基于英伟达图形处理器（GPU）开发运行。

PyTorch是热门开源工具库，开发者与初创企业依靠它搭建、训练和部署人工智能模型。有统计称，85%的人工智能应用使用PyTorch。它最初由Meta在2016年开发，2022年移交Linux基金会管理；因英伟达GPU当时已是研究标准硬件，其诸多功能专门适配英伟达GPU。

TorchTPU可把现有PyTorch应用迁移到TPU，进而迁移到谷歌云。开发者仅需简单修改配置，无需重写应用代码。谷歌还联合Meta、Hugging Face、Inferact、RadixArk等合作，保证TorchTPU与主流开发工具良好兼容。

亚马逊当地时间周一发布类似产品TorchNeuron，可在无需重写代码的情况下把应用移植到Trainium芯片和AWS。亚马逊还称，将向PyTorch主项目提交代码，把功能与英伟达芯片解耦，削弱英伟达对PyTorch的掌控力。

谷歌还推出TPU智能体套件，帮助开发者迁移应用并排查修复错误。其升级了LiteRT工具，以更好支持基于PyTorch的应用；LiteRT用于在手机、笔记本电脑和嵌入式设备上运行人工智能模型。

这些进展不代表开发者会仅因迁移便捷就集体放弃英伟达硬件。但亚马逊和谷歌显然正努力终结英伟达的优势地位。围绕PyTorch的硬件生态竞争升温，芯片与云平台兼容性将成为重要变量。

把任意网站变成干净 JSON

现成爬虫与 AI 智能体

试用 Apify