python - No module named "Torch"

Question

I installed pytorch via

conda install pytorch-cpu torchvision-cpu -c pytorch

And I also tried

pip3 install https://download.pytorch.org/whl/cpu/torch-1.0.1-cp36-cp36m-win_amd64.whl

pip3 install torchvision

Both installed successfully!

But, it only works in jupiter notebook. Whenever I try to execute a script from the console, I get the error message: No module named "torch"

How can I fix this?

Answer 1

Try to install PyTorch using pip:

First create a Conda environment using:

conda create -n env_pytorch python=3.6

Activate the environment using:

conda activate env_pytorch

Now install PyTorch using pip:

pip install torchvision 

_{Note: This will install both torch and torchvision.}

Now go to Python shell and import using the command:

import torch
import torchvision