跳到正文
Google Cloud: Databases·· 3 小时前精选AI 评分67

Google 开源 TorchTPU,PyTorch 可原生运行在 TPU 上

TorchTPU is now open source: Run native PyTorch on TPUs

阅读原文

本站未展示全文,请前往来源网站阅读。

AI 导读

Google Cloud 在 PyTorch Conference 2026 宣布 TorchTPU 作为开源项目发布在 GitHub,TPU 由此成为与 GPU、CPU 并列的 PyTorch 原生选项,由 Google 与 Meta PyTorch 团队共同开发,现有 PyTorch 工作负载只需少量代码修改即可在 TPU 上训练、推理和部署。

推荐理由

TorchTPU 开源后 TPU 成为 PyTorch 原生选项,熟悉的工作流可以少改代码直接迁移,值得评估训练和推理成本。

来源:Google Cloud: Databases · cloud.google.com