-
-
Notifications
You must be signed in to change notification settings - Fork 677
Cannot import torch_tensorrt Python package declared as dependency #2780
Copy link
Copy link
Closed as not planned
Closed as not planned
Copy link
Description
🐞 bug report
Affected Rule
py_binary in our case, but potentially others.
Is this a regression?
Unknown. We did not use torch_tensorrt before.
Description
We are unable to use torch_tensorrt as a Python dependency in our Bazel (version 8.0.1) project. Our py_binary is being built without any issue but on execution of import torch_tensorrt leads to a ModuleNotFoundError.
Importing other packages, like torch works.
🔬 Minimal Reproduction
I created a minimal setup here: https://github.com/scanvio/bazel-torch-tensorrt
🔥 Exception or Error
Traceback (most recent call last):
File "***redacted***/execroot/_main/bazel-out/k8-fastbuild/bin/dummy_torch_tensorrt.runfiles/_main/dummy_torch_tensorrt.py", line 1, in
import torch_tensorrt
ModuleNotFoundError: No module named 'torch_tensorrt'
🌍 Your Environment
Operating System:
Ubuntu 24.04
Output of bazel version:
Bazelisk version: v1.20.0
Build label: 8.0.1
Build target: @@//src/main/java/com/google/devtools/build/lib/bazel:BazelServer
Build time: Fri Jan 17 19:16:16 2025 (1737141376)
Build timestamp: 1737141376
Build timestamp as int: 1737141376
Rules_python version:
1.3.0
Anything else relevant?
Reactions are currently unavailable
Metadata
Metadata
Assignees
Labels
No labels