Cant import torch_tensorrt

Autonomous Machines Jetson & Embedded Systems Jetson AGX Xavier
#1

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

is it problem with the torch from nvidia forum?

#3

Hi,

Could you share your testing script with us?

Based on the doc, there is no torch.fx.passes module.
https://pytorch.org/docs/1.11/fx.html

Thanks.

#4

import torch_tensorrt is not working

#5

Hi,

Do you install it with the instructions shared below?

Torch-TensorRT v1.1.1

Adding support for Torch-TensorRT on Jetpack 5.0 Developer Preview
…

Thanks.