RuntimeError: Expected object of device type cuda but got device type cpu for argument #0 'result' in call to _th_bmm_out

Question

I test my code, but someting wrong. I use the pytorch to load data and test ICP and i have used "cuda()" to add some variables to the GPU. However, the pycharm sho