Error running setup on torch2trt

Sorry for this simpleton problem but I am ramping up on Linux, Python, and all of the above to develop an application for work ( while also doing my day job) …

I am trying to install the torch2trt python module.
I have Pytorch already on my Jetson Nano ( with the AI build)
I follow this procedure:

git clone GitHub - NVIDIA-AI-IOT/torch2trt: An easy to use PyTorch to TensorRT converter
cd torch2trt
python install

But I get the following error:

Traceback (most recent call last):

  • File “”, line 2, in *
  • import torch*
    ImportError: No module named torch

I am new to this environment so I’m sure it’s a moment of stupidity on my part. Please enlighten me.

BTW: I searched for torch module and it shows up in my usr/local/lib/python3.6/dist-packages directory so it’s there. The install went to my home directory. Does this have to do with my intall being in a directory far far away from where Python wants it?


Sorry that the document doesn’t update correctly.
You will need to use python3.6 for the installation:

python3 install


I figured it out a little later, got through that hurdle and now I am looking at :

When I try to verify the install with >>>import torch

Now what am I missing?


Please check this topic for more information about the error: