Pytorchnn only uses one gpu. It's very inefficient when training a large amount of text.
貢獻者指南
技術棧
pythonpytorch
領域
machine learningbackend
議題類型
feature
難度面向新貢獻者的預計實作難度,1 表示很小改動,5 表示專家級工作。
5
預計時間有經驗貢獻者完成調查、實作、測試並準備 pull request 的粗略時間範圍。
over 1 week
活動狀態議題目前的可參與程度:新鮮、活躍、陳舊、阻塞或等待維護者輸入。
stale
清晰度議題是否清楚說明預期改動、驗收標準和下一步。
mostly clear
前置要求
Familiarity with PyTorchUnderstanding of multi GPU trainingExperience with Kaldi's codebaseKnowledge of CUDA
新手友善度1-100 的估計分數,表示該議題對首次貢獻者的友善程度。
15
研究方向
The issue requests extending Kaldi's Pytorchnn module to utilize multiple GPUs. Start by examining the current single GPU implementation in the source code (likely under src/pytorchnn/). Research PyTorch's DataParallel and DistributedDataParallel modules. Check the issue comments for any suggested approaches or pointers from maintainers. Then propose a design that integrates multi GPU support without breaking existing functionality.