I have checked that my tensor is on GPU. The code is too large to be shared here. If you need it, I will try to reproduce it in a small file.
I first want to confirm is there anything else I need to pay attention except adding
ti.cfg.arch = ti.cuda
and putting the tensor and networks to GPU?
BTW, I am on Ubuntu.