一行代码把 PyTorch 搬上 Google TPU:cuda 换 tpu 即可
kuanhoong · x · 2026-11-12
作者分享了一个在 Google TPU 上运行 PyTorch 的小技巧:只需把设备选择从 cuda 改成 tpu 一行代码即可完成迁移,并附上 11 分钟的完整视频教程。
「Infra」频道最新
- vLLM Semantic Router 曝策略绕过漏洞:一个路由头即可跳过安全过滤 — colinmcnamara · 2026-11-13
- GPU 租赁平台 Lium 月度数据:付费客户增 78%,年化收入增 79% — markjeffrey · 2026-11-13
- 传 5090 将成收藏品:Jensen 任内最后一款旗舰显卡? — rachittshah · 2026-11-13
- 黄仁勋亲笔签名的 NVIDIA DGX 到货,Meta 工程师晒图 — iaindunning · 2026-11-13
- 微软专利曝光:闲置时让 AI 用你的 GPU 训练并付钱 — markjeffrey · 2026-11-13
- 曝美司法部对英伟达发起反垄断调查 — GaryMarcus · 2026-11-13