torch.multiprocessing.spawn got error "process 0 terminated with exit code 1"

facebookresearch / moco

PyTorch implementation of MoCo: https://arxiv.org/abs/1911.05722

MIT License

4.83k stars 794 forks source link

Closed miaozhang0525 closed 4 years ago

miaozhang0525 commented 4 years ago

When I use a node with 2 GPUs to run this code, I got this problem. Could you guys help me to solve this?

Sincerely

KaimingHe commented 4 years ago

There is no enough information to locate the problem. I guess if you use 2 GPUs without changing the batch size, it is very likely out of memory.