AGI Hunt· kuanhoong·· 3 小时前AI 评分52
PyTorch 迁移至 Google TPU 仅需将 cuda 改为 tpu
一行代码把 PyTorch 搬上 Google TPU：cuda 换 tpu 即可
AI 导读
作者分享了让 PyTorch 在 Google TPU 上运行的小技巧，只需将设备选择从 cuda 改为 tpu。内容附带一段 11 分钟的完整视频教程，展示相关迁移操作。
来源：AGI Hunt · agihunt.info
PyTorch 迁移至 Google TPU 仅需将 cuda 改为 tpu