Show HN· boxstream·· 27 天前精选AI 评分62
rGPU 发布:让 PyTorch 张量运行在远程 GPU 上
Show HN: Rgpu – a PyTorch device whose tensors live on a remote GPU
AI 导读
rGPU 是一个把 GPU 计算放到远程 NVIDIA 机器、应用留在本地的项目,提供两条接入路径:可选用 rgpu 设备的 PyTorch 程序,以及面向现有 Linux CUDA 程序的 CUDA shim,覆盖 libcuda、CUDA Runtime、cuBLAS、cuBLASLt 和 cuDNN。
推荐理由
项目给出 PyTorch 设备与 CUDA shim 两条接入路径,读者可据此判断远程 GPU 的接入成本与兼容边界。
来源:Show HN · github.com