No module named torch vscode pytorch. This seemed to do the trick for me.