谷歌推出了 TorchTPU，可让 AI 应用在 TPU 和谷歌云上运行 过去，全球绝大多数 AI 算法工程师写的 PyTorch 代码，默认调用的都是 t… · @leo114119 · TwKit

@leo114119

谷歌推出了 TorchTPU，可让 AI 应用在 TPU 和谷歌云上运行 过去，全球绝大多数 AI 算法工程师写的 PyTorch 代码，默认调用的都是 torch.cuda、底层跑的都是英伟达的 cuDNN/TensorRT 算子。 Google 这次推出的 TorchTPU，核心价值在于“即插即用”： 不改代码：在英伟达 GPU 上怎么写 PyTorch，在 TPU 上就怎么写，无需重写底层算子或训练循环； 简单配置：只改几行配置文件，就能直接调动谷歌云上的 TPU 算力集群； 全面开源：谷歌把这套自动翻译、图编译优化的桥梁工具全部开源给社区。 https://cloud.google.com/blog/topics/ai-infrastructure/torchtpu-is-now-open-source-run-native-pytorch-on-tpus

曝光 194 · 评论 1 · 点赞 3 · 书签 1