No module named 'torch'

[kas2ri]

I installed PyTorch as you mentioned. i type import torch in the cmd, its works perfectly. but i try to run train.py in vscode terminal, its shows No module named 'torch'. please help me

[patrickloeber]

make sure your VS Code is using the correct environment

[mrunalmania]

Probably you have setup a multiple python environment in your local PC. Which eventually create this issue.

Solution: Easy: Please install the pytorch from your VS Code's terminal, that will surely solve your problem. Real-Solution:

1. Check how many different Python version you have installed on your PC.
2. Check which Python HOMEPATH is there in your VS Code
3. Try to change it to the one in which you have installed pytorch.

Happy Torching!!!!!!!!!!

[ann-shiku]

make sure your VS Code is using the correct environment

Hello, I am using the correct environment but it's still bringing the same error

[shizhengLi]

In vscode, using Ctrl + Shift + P(windows) to choose a proper Python interpreter.