PyTorch multiprocessing.Pool.map causes segmentation fault

Question

I\'m trying to perform a multiprocessing task on a g4 EC2 instance (32CPUs, 128GB of RAM), using torch.multiprocessing.Pool.

Here\'s my code: