Hi,
Dataloaders by default run all operations in CPU unless you explicitly do otherwise. For instance, you can combine your augmentations using torchvision.transforms then all generated images have been produced in CPU.