ImportError: No module named torch

Question

I'm trying to install this framework from Github called torch2trt (found here:https://github.com/NVIDIA-AI-IOT/torch2trt).

I ran the following code in a Python terminal to install it:

git clone https://github.com/NVIDIA-AI-IOT/torch2trt
cd torch2trt
python setup.py install

but ended up with the following error message:

Traceback (most recent call last):
  File "setup.py", line 2, in 
    import torch
ImportError: No module named torch

I have already installed pytorch using pip install torch torchvision

Does anyone know how to resolve this?

user_13328195 · Answer

Try running everything in python3:

python3 -m pip install torch torchvision

followed by :

python3 setup.py install

user_1513792 · Answer

In Linux,

which python
which pip

In Windows power shell

where python
where pip

Check if both pip and python are from same parent directory.

user_13226586 · Answer

python defaults to python2

So you should change python setup.py install to python3 setup.py install

user_13338285 · Answer

Otherwise you can check the PyTorch page and try another version. https://pytorch.org/get-started/locally/

I would try this one first.

pip install torch==1.6.0+cu101 torchvision==0.7.0+cu101 -f https://download.pytorch.org/whl/torch_stable.html

user_11235205 · Answer

You haven't installed Torch. Install it with pip via: https://pytorch.org/get-started/locally/