Rgpu: 让PyTorch张量驻留远程GPU
Show HN: Rgpu – a PyTorch device whose tensors live on a remote GPU
Rgpu 是一个创新的 PyTorch 设备，让开发者能将张量直接驻留在远程 NVIDIA GPU 上，而无需在本地安装 CUDA。无论是没有 CUDA 支持的 Mac 还是普通笔记本，都能通过 Rgpu 轻松运行高性能深度学习任务。它提供两种集成路径：简单的 PyTorch 设备模式适合新项目，而 CUDA 模拟层则能无缝兼容现有的 Linux CUDA 程序。只需几行代码和 SSH 隧道，即可将本地 Python 应用与远程算力完美连接，彻底打破硬件限制。
让你的 Python 代码留在笔记本电脑上，同时将 PyTorch 运算和张量数据无缝迁移到远程 GPU，即便是没有 CUDA 的 Mac 也能轻松驾驭。