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
Dear experts:
I'm trying to follow your Alexnet example.
I get the following error:
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!