谷歌与亚马逊在PyTorch Conference 2026发布TorchTPU与TorchNeuron开源框架,挑战英伟达CUDA主导地位
谷歌与亚马逊在美国加州圣何塞当地时间10月20日举行的PyTorch Conference 2026上,分别发布开源框架TorchTPU和TorchNeuron,旨在简化现有AI应用从英伟达GPU向自家芯片迁移,挑战英伟达CUDA生态的主导地位。PyTorch由Meta(原Facebook)的AI研究实验室FAIR于2016年主导开发,全球约85%的AI应用开发者使用它,而FAIR最初围绕英伟达GPU和CUDA平台设计PyTorch,导致向其他芯片迁移困难。谷歌与Meta合作开发的TorchTPU,据谷歌称,超过98%的现有PyTorch模型只需将代码中的设备字符串改为model.to("tpu")即可在TPU上运行,转换耗时从几分钟到几天不等,且TPU已在PyTorch核心层面作为与GPU、CPU同等的原生设备运行。AWS的TorchNeuron则允许基于PyTorch的AI模型直接运行在AWS的Trainium AI芯片上,开发者只需修改几行设备设置,把英伟达的cuda替换为AWS的neuron。亚马逊已通过GitHub以Apache 2.0开源许可发布TorchNeuron公测版,并向PyTorch主开源项目直接贡献代码。此前将针对英伟达生态优化的AI应用迁移到谷歌或亚马逊专有AI芯片至少需数天,复杂情况下需数月,新工具有效消除了这些时间和成本障碍。随着TorchNeuron、TorchTPU及类似工具获得采用,专用AI芯片预计将主要在底层性能、能效和定价上竞争,但针对具体模型的兼容性、性能和成本验证仍需额外时间。
- Related manufacturers
- Google, Amazon, Meta, Nvidia, Amazon Web Services (AWS)
- Impact score
- 55 / 100, medium impact