[BUG]: ModuleNotFoundError: No module named 'tensorrt' when using morpheus tools #2076
Closed
2 tasks done
Labels
bug
Something isn't working
Version
24.10
Which installation method(s) does this occur on?
Conda
Describe the bug.
This appears to be a repeat of #1979, It looks like tensorrt-cu12 v10.7 might be causing this.
Minimum reproducible example
conda env update --solver=libmamba -n morpheus --file conda/environments/model-utils_cuda-125_arch-x86_64.yaml cd models morpheus --log_level=info tools onnx-to-trt --input_model sid-models/sid-minibert-20230424.onnx --output_model ./model.plan --batches 1 8 --batches 1 16 --batches 1 32 --seq_length 256 --max_workspace_size 16000
Relevant log output
Click here to see error details
Full env printout
Click here to see environment details
Other/Misc.
No response
Code of Conduct
The text was updated successfully, but these errors were encountered: