Hacker News AIT2·68 分
开源 rGPU:实现 PyTorch 张量远程 GPU 执行与 CUDA 兼容层透传
原文标题:Show HN: Rgpu – a PyTorch device whose tensors live on a remote GPU
- 支持两种模式:原生 PyTorch 设备接口(简单集成)和 CUDA Shim(兼容现有二进制文件)。
- 通过 TCP 协议传输操作,实现应用逻辑与 GPU 计算的物理分离。
- 包含 nanoGPT 等示例及完整的运维、性能调优文档,具备生产可用性潜力。