No module named torch vscode. Give this a try and see if it resolves the issue.