Would TorchData provide GPU support for loading and preprocessing images?

[songyuc]

🚀 The feature

Would TorchData provide GPU support for loading and preprocessing images?

Motivation, pitch

When I am learning PyTorch, I find, currently, it do not support using GPU to load images or any other transforms of preprocessing and encoding data. I want to know whether this would be taken into consideration into the design of TorchData.

Alternatives

Currently, NVIDIA-DALI is an impressive alternative for loading and preprocessing images with GPU.

Additional context

No response

[msaroufim]

Hi @songyuc this is something I'm working on and will keep you posted on plans

[songyuc]

Hi @songyuc this is something I'm working on and will keep you posted on plans

I am looking forward to your contribution sincerely🥳

[Atharva-Phatak]

@msaroufim I am interested to help you out if its possible. Seems pretty useful :). Let me know if I can do anything here.

[songyuc]

Just for discussion, would torch.cuda.Stream facilitate GPU data loading?

[wenleix]

cc @dracifer