Reduce the num_workers used when you create your dataloader.
num_workers (int, optional) – how many subprocesses to use for data loading. 0 means that the data will be loaded in the main process. (default: 0)
https://pytorch.org/docs/stable/data.html#torch.utils.data.DataLoader 70