No module named torch vscode. It is recommended to use Python 3.