ERROR:
from torch.fx.passes.pass_manager import PassManager
ModuleNotFoundError: No module named ‘torch.fx.passes.pass_manager’
PyTorch Version : 1.11
CPU Architecture: jetson AGX xavier
OS (e.g., Linux):
How you installed PyTorch: nvidia forum wheel
Build command you used (if compiling from source):
Are you using local sources or building from archives:
Python version:3.8
CUDA version: 11.4
Hello! I am running Jetpack 5.0.2 on a Jetson Xavier NX and have the same problem as the original poster here. I installed Pytorch from the repository provided by NVIDIA for Jetson devices and installed Torch TensorRT in line with the instructions specific to Jetson devices. Any help would be much appreciated.
Traceback (most recent call last):
File "<stdin>", line 1, in <module>
File "/home/nvidia/.local/lib/python3.8/site-packages/torch_tensorrt/__init__.py", line 85, in <module>
from torch_tensorrt._compile import *
File "/home/nvidia/.local/lib/python3.8/site-packages/torch_tensorrt/_compile.py", line 9, in <module>
import torch_tensorrt.fx
File "/home/nvidia/.local/lib/python3.8/site-packages/torch_tensorrt/fx/__init__.py", line 12, in <module>
from .lower_setting import LowerSetting # noqa
File "/home/nvidia/.local/lib/python3.8/site-packages/torch_tensorrt/fx/lower_setting.py", line 5, in <module>
from torch.fx.passes.pass_manager import PassManager
ModuleNotFoundError: No module named 'torch.fx.passes.pass_manager'
I’m having the same problem as @wesley3 .
The proposed solution doesn’t work for me. We are using TensorRT with Jetpack 5 on a Nvidia Xavier AGX.
Any help would be much appreciated.