Well, there is no error when I remove the .cuda() parts. I think it runs properly in CPU. What do you recommend? I am barely sure that the index is not a problem because I can run it on another laptop which has GTX 1660. Same code, no error