Hi.
I am facing a strange problem. When running a code on two GPUs my machine, after some epochs, reboots. If I use the same code running only on 1 GPU everything works fine.
I have follow this old thread nn.DataParallel(model).cuda() stuck 7 but it does not work.
Any suggestions?.