TorchTPU 迁移实战：3 个 CUDA 卡点我把手上一个典型的 GPU 训练工程搬到 TorchTPU 上试了一遍，「换芯片要重写几个月代码」这句话，在 2026 年 10 月 20 日之后需要打个折。那天 PyTorch Conference 上，Google 与 Meta 的 PyTorch 团队一起把 TorchTPU 开源了：TPU 成为一个原生 PyTorch 设备，理论上你只需要把 .to("cuda") 改成 .to("tpu")。我把手上一个典型的 GPU 训练工程拿去试了一遍，改一行是真的，跑起来之后才发现问题都在那一行之外。这篇