Training crashes due to - Insufficient shared memory (shm) - nn.DataParallel

This solved my problem! Thank you