from torch2trt.module_test import add_module_test ModuleNotFoundError: No module named 'torch2trt.module_test'

[KatBendtz]

Dear experts:

I'm trying to follow your Alexnet example.

I get the following error:

from torch2trt import torch2trt

File "/usr/local/lib/python3.8/dist-packages/torch2trt/torch2trt/init.py", line 2, in from .converters import * File "/usr/local/lib/python3.8/dist-packages/torch2trt/torch2trt/converters/init.py", line 7, in from .AdaptiveAvgPool2d import * File "/usr/local/lib/python3.8/dist-packages/torch2trt/torch2trt/converters/AdaptiveAvgPool2d.py", line 2, in from torch2trt.module_test import add_module_test ModuleNotFoundError: No module named 'torch2trt.module_test'

My system: Ubuntu 20.04 Cuda 11.3 TensorRT 8.4 x86/64

I installed through:

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

When I execute the script inside the torch2trt folder it works.

Very grateful for any help!

[XiantaoHu]

I have the same question