GPU transform

[hw-ju]

In the output of python monai/transforms/utils.py, a transform can run on GPU if it's 1 under Torch? , correct?

[Nic-Ma]

Hi @hw-ju ,

Yes, you can convert data from numpy array to torch Tensor and run on GPU, refer to this example:
https://github.com/Project-MONAI/tutorials/blob/master/acceleration/distributed_training/brats_training_ddp.py#L198-L199

Thanks.

[hw-ju]

Thanks @Nic-Ma!
Just to confirm, only torch Tensor can be run on GPU, numpy array can't, correct?

[rijobro]

Correct. In theory we could possibly use cupy for some of these but it's not something we've explored in great detail yet. Which transform is in numpy that you're hoping to do on the GPU?

[hw-ju]

Thanks @rijobro! I don't have those numpy transforms on my mind now, but will report here when they come up. :D