search

NVIDIA-AI-IOT / torch2trt

An easy to use PyTorch to TensorRT converter
MIT License
4.55k stars 671 forks source link

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

Open KatBendtz opened 2 years ago

KatBendtz commented 2 years ago

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 commented 2 years ago

I have the same question