谷歌要釜底抽薪英伟达 TorchTPU 让 pyTorch只改一行代码就能从cuda迁到TPU。 谷歌开源TorchTPU。 现有PyTorch代码把.to… · @Zen_with_AI · TwKit

@Zen_with_AI

谷歌要釜底抽薪英伟达 TorchTPU 让  pyTorch只改一行代码就能从cuda迁到TPU。 谷歌开源TorchTPU。 现有PyTorch代码把.to("cuda")改成.to("tpu")就能在云TPU上训练和部署了，几乎不用重写。这是谷歌与Meta联合开发的原生后端，v5e到v7的芯片都已支持。 https://discuss.google.dev/t/torchtpu-and-the-tpu-agent-suite-are-now-open-source-for-native-pytorch-execution-on-google-tpus/407405

曝光 160 · 评论 0 · 点赞 3 · 书签 1