跳到正文
原文
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