torch.multiprocessing.set_sharing_strategy('file_system') Using this solved for me.
torch.multiprocessing.set_sharing_strategy('file_system')