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 setup.py install
But I get the following error:
Traceback (most recent call last):
- File “setup.py”, 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?