Uh, no, I’m working in vsc, vsc gives me that error.
It is working in jupyter. I mentioned colab because thought maybe torch.device was changed to something else in new update.
EDIT
I dont think so that i messed up something, since jupyter is not giving me any errors and + its fresh new install