ModuleNotFoundError: No module named 'torch'

[venkat-ranganathan]

Hi,

I’m installing the JetRacer dependencies on my Jetson Orin Nano dev kit running JetPack 5.1.3

I am having issues installing torch2trt from: jetracer/docs/software_setup.md at master · NVIDIA-AI-IOT/jetracer · GitHub

I had no issue installing JetCam, but I keep getting the modulenotfound error for torch when trying to install torch2trt

I followed the instructions here to install PyTorch: Installing PyTorch for Jetson Platform - NVIDIA Docs

I can run $ python3, then >> import torch and I get no errors

but when I try to install torch2trt I still get the following error:

How can I resolve this? I do not want to use a docker container

[AastaLLL]

Hi,

Please install PyTorch with the instructions below:

Installing PyTorch for Jetson Platform - NVIDIA Docs

Thanks.

[venkat-ranganathan]

Hi,

Those are the instructions I followed, I linked the same instructions in my original post in the 4th paragraph

The error I get is after following the instructions from that link

Where does torch2trt expect PyTorch to be installed?

[venkat-ranganathan]

Hi,

I resolved the error. I needed to set the --target to install in /usr/local/lib/python3.8/dist-packages/torch it was installing in a different location

format is this:
python3 -m pip install package_name --target /path/to/custom_directory

This thread helped: Error while installing torch2trt on Xavier NX - #7 by AastaLLL