@JiangPQ some notes:

instead of installing Pillow, you can try pip install pillow-simd, it is faster for data-augmentation by a significant amount
you can give this dataloader a try, it was written for HDD:  https://github.com/BayesWatch/sequential-imagenet-dataloader 494