跳到正文
Hacker News (官方 RSS)·· 2 天前精選AI 評分77

rGPU:PyTorch 遠端 GPU 介面

Show HN: Rgpu – a PyTorch device whose tensors live on a remote GPU

AI 導讀

rGPU 是一個可在遠端 NVIDIA 伺服器上執行 GPU 工作的 PyTorch 裝置,使用者仍在本機執行程式。

  • 兩種整合方式: - PyTorch 裝置:透過 torch.device('rgpu') 直接呼叫,安裝簡單(pip install rgpu)。
推薦理由

原文說明 rGPU 讓 PyTorch 程式可遠端執行 GPU,讀者可評估此方案對分散式訓練的可行性。

來源:Hacker News (官方 RSS) · github.com