Solved by heavyfranz in post #4 
              
              
                Any news? Have you solved the problem? How? I think that the heart of @bapi answer is that you have to manually transfer each input array (a fraction of it or the same, it depends on your problem) 
I solved like this: 
import time
import torch
from torch.multiprocessing import Pool
torch.multiproces…