Will you implement image transforms on DALI or in GPU?

[ZhongkangLu]

Hi,

I am working on 3D MRI image segmentation. The augmentation in CPU is quite slow and my GPUs have to wait for it. Will you implement image transforms in GPU? Or base on the tools such as Nvidia DALI?

Thanks

[Nic-Ma]

Hi @ZhongkangLu ,

Most of the MONAI transforms can execute on GPU directly, please check this documentation and the tutorial:
https://github.com/Project-MONAI/tutorials/blob/main/acceleration/fast_model_training_guide.md#2-execute-transforms-on-gpu

Thanks.

[ZhongkangLu]

Thank you fr your prompt response! Really appreciated.