一行代码把 PyTorch 搬上 Google TPU:cuda 换 tpu 即可

kuanhoong · x · 2026-11-12

作者分享了一个在 Google TPU 上运行 PyTorch 的小技巧:只需把设备选择从 cuda 改成 tpu 一行代码即可完成迁移,并附上 11 分钟的完整视频教程。

原文链接 →

「Infra」频道最新

更多「Infra」频道 AI 资讯 →