pytorch/examples

DDP training question

Open

#1.096 geöffnet am 25. Nov. 2022

Auf GitHub ansehen
 (2 Kommentare) (0 Reaktionen) (1 zugewiesene Person)Python (21.634 Stars) (9.429 Forks)batch import
distributedhelp wanted

Beschreibung

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?

Contributor Guide