pytorch/examples

DDP training question

Open

#1,096 opened on 2022年11月25日

GitHub で見る
 (2 comments) (0 reactions) (1 assignee)Python (21,634 stars) (9,429 forks)batch import
distributedhelp wanted

説明

Hi, I'm using the tutorial https://github.com/pytorch/tutorials/blob/master/intermediate_source/ddp_tutorial.rst for DDP train,using 4 gpus in myself code, reference Basic Use Case. But when I finished the modification, it was stuck during run the demo,meanwhile,video memory has been occupied.Could you help me?

コントリビューターガイド

技術スタック
python
領域
machine learningbackend
Issue 種別
bug
難度新しい貢献者にとっての実装難度です。1 は非常に小さな変更、5 は専門的な作業です。
5
推定時間経験ある貢献者が調査、実装、テスト、pull request 準備にかけるおおよその時間範囲です。
1-3 hours
活動状況Issue が今どれだけ取り組みやすいかを示します。新しい、活発、古い、ブロック中、またはメンテナー入力待ちなどです。
stale
明確さ期待される変更、受け入れ条件、次の手順がどれだけ明確かを示します。
mostly clear
前提条件
PyTorch basicsDistributed training conceptsCUDA environment
初心者向け度初回貢献者にどれだけ取り組みやすいかを 1-100 で推定したスコアです。
20
調査方針
The user is stuck while running DDP training with 4 GPUs based on a PyTorch tutorial. The issue may arise from version incompatibility or misconfiguration. The assignee should check the referenced tutorial file and provide guidance. More details about the error logs are needed.