《Born》第12章：GPU 训练实战——从 CPU 到 GPU 的迁移

把第 11 章的 MNIST 训练从 CPU 迁移到 GPU，只需要改一行代码。

一行代码切换后端

// CPU 版本
backend := cpu.New()

// GPU 版本
backend := webgpu.New()

就这一行。模型定义、训练循环、数据加载——全部不用改。

性能对比

内存注意事项

GPU 内存比 CPU 内存紧张得多：

// 监控 GPU 内存使用
backend := webgpu.New()
backend.SetMemoryBudget(4 * 1024 * 1024 * 1024) // 4GB 预算

当接近预算上限时，Born 自动触发内存回收。

混合精度训练

// FP16 训练（节省显存，加速计算）
backend := webgpu.New(webgpu.Config{Precision: webgpu.FP16})

WebGPU 原生支持 FP16 运算。

📘 《Born》连载技术书，第 12/22 章。